过程:
一、创建虚拟环境之后,按照官网:
Previous PyTorch Versions | PyTorch
找到想要安装的版本1.8.0
使用命令行安装:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
二、用conda list 检查版本
发现安装的确实是1.8.0
三、测试
import torch
print(torch.cuda.is_available())
print(torch.__version__)
输出显示的却是1.3.0+cpu版本
conda list | grep torch
也没有发现1.3.0版本的torch出现
四、解决方法
pip uninstall torch
关掉再试试就好了