建议全文看完再操作
查看显卡
1.查看自己gpu的型号
ctrl+Alt+delete 打开任务管理器,点击性能,
2.查看cuda是否支持自己电脑的gpu型号
进入NVIDIA的官网,罗列出来的就是支持的型号啦
https://www.geforce.com/hardware/technology/cuda/supported-gpus
3.查看driver version 和cuda version
这一步很重要哦,之后torch.cuda.is_available()返回false的主要原因
打开ananconda promote,输入
nvidia-smi
cuda 9.2以上是支持驱动版本大于392.26的,如果小于392.26,就要升级,可以借助于各种电脑管家升级,也可以进入英伟达的驱动程序下载界面
https://www.nvidia.com/Download/index.aspx?lang=cn
在官网选择相应的选项后,红色框部分,默认即可。产品系列中的 Notebooks 指的笔记本。将下载后的驱动,正常安装即可。
anaconda中新建环境
conda create --name pytorchgpu python=3.7
启动新环境
activate pytorchgpu
安装清华镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
``
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
进入pytorch官网选择对应的命令
https://pytorch.org/get-started/locally/
换源安装
去掉官网给出的指令 -c后面的所有内容
conda install pytorch torchvision cudatoolkit=10.1
ps:某一天下午下载的时候老是遇到网络问题,如下图,下到一般就崩了
建议:早上的时候网络比较通畅,再去安装清华镜像,再去下载,下载的过程中就只有pytorch这一项的进度条在慢慢的动,500多兆,下载完成之后,其他的包也都在了。
测试
在测试之前建议先重启anaconda promote
activate pytorchgpu
python
import torch
import torchvision
torch.cuda.is_available()
but!!依旧返回false啊!
于是,我想到了,cuda的版本匹配问题,
输入nvidia -smi 后看到cuda version 是10.0,
实际操作:
第一次在官网下载选择的cuda 版本是10.1,返回false
第二次在官网下载选择的cuda 版本是10.0,返回ture 了。而且下载的时候贼快3分钟左右就OK了,但是用清华镜像不能下载9.2版本的,报错说是当前镜像找不到9.2的包。
所以,版本的选择还是要跟自己的电脑上的cuda version匹配吧,不能高于自己电脑上的cuda version
第二次新建环境 pytorch_gpu
运行
conda install pytorch torchvision cudatoolkit=10.0
3分钟解决
以上整理的内容是参考了
https://www.bilibili.com/video/BV1nE411b7Vx
看完视频之后,安装还是有问题,经过探索和跟up主讨论之后,安装成功了。
附图: