安装Pytorch后torch.cuda.is_available()返回False问题解决
首先说几点安装Pytorch时需要注意的点:
1.如果有能力,最好在科学上网后,利用类似如下命令直接安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
这里注意:上述命令行最后没有-c pytorch,这点与官网的不一样,官网给出的是这样的:
加上-c pytorch的意思好像是从清华源下载~~
2.如果能力不足,无法科学上网,则需要手动切换镜像源了,比如切换清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn
之后仍然利用下述命令安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
注意:这里还是没有 -c pytorch。因为我加上 -c pytorch后发现安装会有问题,所以还是去掉 -c pytorch。
3.如果要安装GPU版本的Pytorch,则需要你的电脑上有NVI
安装Pytorch后torch.cuda.is_available()返回False问题解决最先出现在Python成神之路。
共有 0 条评论