下载旧版本torch 以及选定对应的torchision
pip3 install torch==0.4.1 torchvision==0.2.1 -i https://pypi.douban.com/simple
指定torch和对应torchvision的版本,并通过豆瓣镜像下载,这样会更快。(偶尔清华的镜像也挺快的,但是今天依旧read time out 了。
删除之前的旧版本
sudo pip3 uninstall torch
这里有个问题:不能确定cuda10能不能与pytorch0.4版本兼容(有结果了,如下图和参考后面多个链接)
!!!!!!
图一的问题在于旧版本的pytorch不能加载高版本的模型结果。参考链接1
图二的问题在于cuda10与低版本的torch不对应造成某些程序无法执行。参考链接2.0,链接2.1
新的坑,填一填:
环境cuda9.0,(对应cudnn7.3)下载torch版本1.0.0,必须下载torchvision0.2.0,否则必会报错,报错内容很奇葩。
如果下载torchvision0.3.0就会自动下载torch的最新版本,而且这个版本在此环境和torch1.0.0版本下也会报错。