由于要用一些奇怪的包,导致需要安装与cuda版本匹配的pytorch,然而:

cuda  python教程 cuda9.1 pytorch_服务器


惊喜地发现实验室服务器上的cuda版本是9.1的。但是又不可能去更新服务器的cuda版本,所以只能硬着头皮去找看看有没有旧版的pytorch。而这个时候去pytorch官网找的话:

cuda  python教程 cuda9.1 pytorch_服务器_02


有cuda9.0的,有9.2的,就是没有9.1的。这个时候其实可以打开:

https://download.pytorch.org/whl/cu91/torch/

这里面其实就有支持cuda9.1的pytorch包:

cuda  python教程 cuda9.1 pytorch_服务器_03


可以看到最多只支持到pytorch 0.4.0(cuda 9.1有点老了),找到适合自己的包利用pip进行安装即可,例如:

pip install https://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-linux_x86_64.whl