算力3.5,现在绝大多数笔记本的独显都有了,我的是tesla k40m,这里提供如何安装cuda版本的torch,如果需要tensorflow也一起装了,下面是我的配置,供大家参考:
这几个版本一定要对应好:
python=3.6
cudatoolkit=9.0
cudnn=7.6.4
tensorflow-gpu==1.10.0
torch-1.2.0+cu92-cp36-cp36m-manylinux1_x86_64.whl
torchvision-0.4.0+cu92-cp36-cp36m-manylinux1_x86_64.whl
numpy==1.15.1
mkl-fft==1.0.6
pandas==1.0.0 # 一定要指定这个版本,否则默认装最新的pandas会升级numpy,然后tensorflow和torch就不可以用了
1. windows的话,清华源下载Anaconda3-2019.07-Linux-x86_64.sh,然后rz到服务器上(rz 是linux的一个指令,你在xshell中敲一下就知道干什么的了,为什么在windows下载,因为我这边服务器网速太慢),然后sh Anaconda3-2019.07-Linux-x86_64.sh一路到头,有yes/no输入yes,部分版本会问你是否需要安装VSCode,输入no;(如果清华源也很慢,就用北外源,https://mirrors.bfsu.edu.cn/anaconda/archive/)