cuda和cudnn
pytorch和torchvision版本没对应上
pytorch和torchvision版本对应关系
以下版本截止2022年5月5日
pytorch torchvision python...0.5.0 ==2.7, >=3.5, <=3.8 9.2, 10.0
1.3.1 0.4.2 ==2.7, >=3.5, <=3.7 9.2, 10.0
1.3.0 0.4.1 ==2.7, >=3.5...2.7, >=3.5, <=3.7 9.0, 10.0
conda安装方法
根据pytorch官网方法安装,参考
https://pytorch.org/get-started...9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch
# CUDA 10.0
conda install...torchvision==0.4.0 cpuonly -c pytorch
后话
其实,环境中各版本对应关系出错,很容易导致各种乱七八糟地错误,所以尽量按照官方的要求安装。