查看自己的显卡支持的CUDA版本

  1. 在开始中找到并打开NVIDIA控制面板。
  2. 打开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
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐