问题: 显示torch.cuda.is_available()=False
解决办法:说明这个虚拟环境不可用,删除虚拟环境,重建一个新的虚拟环境
1、删除原来的虚拟环境,假如原虚拟环境为pytorch-old,输入以下命令,先退出当前虚拟环境,然后删除原虚拟环境
conda activate base
conda remove -n pytorch-old --all
2、在pycharm终端输入 conda create -n pytorch 创建一个新的名为pytorch的虚拟环境
conda create -n pytorch
3、在pytorch官网(https://pytorch.org/)上找到符合自己的安装命令,以下是我的安装命令
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
4、安装成功后检查当前的cuda是否可用执行以下命令:
python
import torch
torch.cuda.is_available()
成功: