运行github上的代码,经常人家的环境是比较老的cuda和torch,这时候使用新版本会导致其余依赖库的安装失败,这篇记录一下如何安装老版本的torch(重新搞cuda太麻烦了,后续可以使用docker镜像,但我暂时不会):
去这个地方找需要的cuda版本对应的torchwhl文件:
https://download.pytorch.org/whl/torch_stable.html
一定要看好了下载,不要下载成cpu。如图是下载cuda为11.0,torch为1.7,python为3.6在linux系统下的文件
ubuntu系统基本是会给放到downloads文件夹下面,之后激活虚拟环境再下载即可,使用如下命令,把后面的文件名换了就行:
pip install ~/downloads/torch-1.7.0+cu110-cp37-cp37m-linux_x86_64.whl
pip install ~/downloads/torchcsprn-0.2.0+cu110-cp37-cp37m-linux_x86_64.whl
pip install ~/downloads/torchvision-0.8.0-cp37-cp37m-linux_x86_64.whl