textgenrnn ubuntu 21.04

textgenrnn_ubuntu2104.sh

conda create --name textgenrnn python
conda activate textgenrnn

conda install cudatoolkit==10.2.89
# this special conda source is vital to get the correct 8.1.x version of cudnn:
conda install -c conda-forge cudnn

git clone https://github.com/minimaxir/textgenrnn.git
cd textgenrnn
pip install -r requirements.txt
# tf version error here expected, carry on.
pip install tensorflow-gpu
pip install tqdm sklearn

# this is going to sound weird, but cusolver 11 doesn't exist yet so we fake it:
sudo ln -s ~/anaconda3/pkgs/cudatoolkit-11.0.221-h6bb024c_0/lib/libcusolver.so.10  ~/anaconda3/pkgs/cudatoolkit-11.0.221-h6bb024c_0/lib/libcusolver.so.11

# install from  repo so you can import anywhere
python setup.py install

# mod this line as neccesary:
export LD_LIBRARY_PATH=/home/dhlab/anaconda3/pkgs/cudatoolkit-11.0.221-h6bb024c_0/lib