第一部分:问题描述
我的安装命令为:
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
在安装pytorch的时候出现如下错误信息:
第二部分:解决方法
我的解决办法是:
pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
也就是在最后面加上下载源。
同时需要注意,torch1.2.0版本是比较早的版本了,所以这个默认是需要python3.7才可以的,如果你是3.8的话,换源也是没用的!
可以发现已经开始正在下载了!
下面我们开始验证版本:
在命令提示符中输入以下命令进入 Python 交互模式:
python
验证torch:
import torch
print("Torch version:", torch.__version__)
1.2.0
:表示 PyTorch 的主版本号,具体是 PyTorch 1.2.0。+cu92
:表示 PyTorch 是针对 CUDA 9.2(cu92
)编译的版本
验证torchvision:
import torchvision
print("Torchvision version:", torchvision.__version__)
这个错误是因为 torchvision
0.4.0 版本依赖于旧版本的 Pillow
库,而你安装的 Pillow
是 9.5.0 版本,新版本的 Pillow
中不再使用 PILLOW_VERSION
这个属性,导致了导入失败
在命令行中运行以下命令来降级 Pillow:
pip install pillow==6.2.2
再次运行:
版本正确显示!