安装GPU加速的TSNE

安装GPU加速的TSNE

conda create -n gpu_tsne python=3.8

conda install tsnecuda -c conda-forge

测试是否安装成功：

import tsnecuda
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
tsnecuda.test()

来自：

https://github.com/CannyLab/tsne-cuda/blob/master/INSTALL.md

安装指定cuda版本

参考：https://fangdahan.medium.com/install-and-use-tsnecuda-package-a1355902eb90

1. 假设已经安装了python=3.8, cudatoolkit=10.2，然后根据https://anaconda.org/CannyLab/tsnecuda/files了解应该安装linux-64/tsnecuda-3.0.0-h1234567_py3.8_73_cuda10.2.tar.bz2可以直接下载下来离线安装。