双显卡电脑,安装pytorch,输入,
import torch
torch.cuda.is_available()
显示FALSE。
首先要排除的原因,
从官网使用
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
安装的torch是否是GPU版本。
解决方法:https://download.pytorch.org/whl/torch_stable.html
打开链接安装相对应的torch和torchvision版本。复制到安装的anaconda安装目录
然后在使用官网的安装代码 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
之后按照响应的安装步骤安装即可。
其次,查看anaconda以及pytorch的版本是否匹配,不行就全部安装最新版本。