可以import torch但torch.cuda.is_available()返回false

参考:
torch.cuda.is_available()返回false——解决办法
完美解决torch.cuda.is_available()一直返回False的玄学方法
我的最终解决办法:
到https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/上找到对应的pytorch,torchvision,torchaudio版本(非cpu版本),尽量对上号,下载,放入annaconda的pkgs文件夹中,然后 conda install 指令安装
接下来再使用对应的conda命令,比如

conda install pytorch1.10.0 torchvision0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

这时候就不会从清华源下载cpu版本的了,还能补全剩余的所需要的package。

你可能感兴趣的:(pytorch)