GeForce RTX 2060配置CUDA并安装pytorch
查看自己的显卡支持的CUDA版本
- 在开始中找到并打开NVIDIA控制面板。
- 打开NVIDIA控制面板。选择“帮助”–“系统信息”–“组件”,找到NVCUDA.DLL信息显示即为显卡支持的CUDA最高版本。
根据支持的CUDA版本去下载对应的VS,安装VS2017
1.1现在VS官网主推的是2019版本,2017版本的安装包。
链接:https://pan.baidu.com/s/1m7N1A7HMZdbTVJ22UhkEUg
提取码:jm3o
按照默认安装即可,安装过程中,把.NET和C++开发都勾选上。
1.2 安装Cuda
在安装过程中,会自动检测本机是否已经安装了配套的VS版本,如果VS版本和Cuda版本不匹配的话,安装无法进行。
( 另外,如果电脑安装了360杀毒的话,安装过程中会不断有疑似病毒修改的提示,要全部允许操作,否则无法安装。)
1.3 检验是否安装成功
nvcc --version看是否可以查看到版本号
安装pytorch
- 在安装pytorch时,根据自己的版本选用https://pytorch.org/官网上的安装命令。但这样是从国外下载,很慢且容易中途挂掉或出错。
- 解决方法:
第一步、添加镜像站到Anaconda执行如下命令:
conda config --add channels http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
第二步、切记在官网的命令上去去除:-c pytorch
conda install pytorch torchvision cudatoolkit=10.0