ubuntu上pip安装pytorch1.1.0和torchvision0.4

官网找到相应的pytorch和torchvision版本并下载whl文件到本地,最新版conda安装的话首页直接复制命令就行:

ubuntu上pip安装pytorch1.1.0和torchvision0.4_第1张图片

其他版本需要进入以下链接:

https://download.pytorch.org/whl/cu100/torch_stable.html

ubuntu上pip安装pytorch1.1.0和torchvision0.4_第2张图片

ubuntu上pip安装pytorch1.1.0和torchvision0.4_第3张图片

选择想要的pytorch和torchvision版本,cp36对应python3.6,复制链接地址下载。

进入文件保存的目录,运行pip3 install torchvision-0.4.0-cp36-cp36m-manylinux1_x86_64.whl torch-1.1.0-cp36-cp36m-linux_x86_64.whl即可:

貌似torch-1.1.0对应的torchvision应该是0.3.0版本,torchvision0.4.0版本对应pytorch1.2及以上。但之前装的torchvision-0.3.0跟cuda版本冲突,import报错:ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory。(我安装tensorfow的时候cuda装的是10.0,当然找不到9.0)所以卸载重装了torchvision0.4.0,目前没啥问题,先就这样了。

 

 

你可能感兴趣的:(pytorch)