GeForce RTX 2060配置CUDA并安装pytorch

查看自己的显卡支持的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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值