ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Win10系统配置RTX3060ti GPU深度学习主机pytorch环境

2021-01-06 13:29:04  阅读:2712  来源: 互联网

标签:RTX3060ti 版本 cuDNN pytorch CUDA https Win10 安装 下载


Win10系统配置RTX3060ti GPU深度学习主机pytorch环境

0. 前言

前不久英伟达发布了30系显卡,受到了各位炼丹师的关注。据说3060ti性能媲美2080super,其性价比极高(涨价之前)。但新显卡也有缺点,那就是不兼容旧版本的CUDA,只支持11.0以上的版本。pytorch也紧随潮流,正式支持CUDA11.0版本。

在这里插入图片描述
以下是我亲测有效的使用 GPU 3060ti 的各部分安装版本

电脑系统:window 10
python版本:3.8
pytorch版本:1.7.1
CUDA版本:11.0
cuDNN版本:8.0.5 适用于CUDA11.0

如果是刚入门的朋友,请看以下安装教程,手把手教学,大神请随意~

1. 安装Anaconda

推荐直接上官网下载
https://www.anaconda.com/products/individual
如图所示,下载64位windows版本。(现在应该没有用32位的吧~)
在这里插入图片描述
下载完毕后双击exe文件的图标打开安装界面,如图所示。
在这里插入图片描述
在这里插入图片描述
按照指示安装完毕即可。

***** 新建环境 *****

这里介绍通过Anaconda软件内通过鼠标点击UI来创建环境。

另外,还可以命令提示符来创建环境,网上教程很多,这里就不赘述了。
在这里插入图片描述

2. 安装CUDA和cuDNN

(1) 安装显卡驱动

首先需要下载和安装显卡驱动,进入下面网址选择合适的显卡驱动下载并按照提示安装。如果已安装此处略过。
https://www.nvidia.cn/geforce/drivers/
在这里插入图片描述

下载CUDA不需要注册,但下载cuDNN需要注册和登录。

(2) 下载CUDA

CUDA下载网站
https://developer.nvidia.com/cuda-toolkit-archive

选择需要的版本,这里选了11.0版本,因为3060ti只支持11.0以上版本,而pytorch当前最高支持11.0版本。
在这里插入图片描述
选择WIN10、本地安装,再点击下载安装包即可。
在这里插入图片描述

(3)下载cuDNN

cuDNN网址
https://developer.nvidia.com/cuDnn

下载cuDNN需要注册NVIDIA账号并登录,注册时安装官网提示填写用户名、密码、邮箱等,其中需要邮箱验证。

邮箱验证时使用QQ邮箱接收不到验证邮件,使用163邮箱可以。

邮箱验证完后,补全一些信息即可。
在这里插入图片描述
随后进入以下页面,选择使用CUDA11.0、Windows的cuDNN库,点击下载。
在这里插入图片描述

(4) 安装CUDA和cuDNN

建议安装CUDA之前先安装VS,可选VS2013,体量小。

打开CUDA安装包。
在这里插入图片描述
在这里插入图片描述
安装CUDA完毕后,解压cuDNN压缩包,复制以下所有内容。
在这里插入图片描述

打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

在这里插入图片描述
将刚才复制的内容粘贴到该文件夹。
在这里插入图片描述
完成。

3. 下载Pytorch

这里有两种下载方式,推荐第二种。

下载方式1(慢,易失败)

打开Pytorch官网:
https://pytorch.org/get-started/locally/
在这里插入图片描述
选择好版本后,复制蓝框内的代码到命令窗口自动下载,但这种下载方式极慢,且容易下载失败,失败后需要重新下载。有vpn的朋友可使用这种方法。

下载方式2

进入如下网址
https://download.pytorch.org/whl/torch_stable.html
找到对于版本的文件直接下载。这种下载方式速度快多了
在这里插入图片描述
在这里插入图片描述

安装方式

使用管理员权限打开命令窗口。

  • 激活环境,activate [环境名]
  • 去到下载好的文件夹路径
  • 用图中的命令安装
    在这里插入图片描述
    如果途中提示需要安装其他库,但是又安装得特别慢,可以尝试使用别的镜像源,例如:
pip install numpy==1.19 -i https://pypi.douban.com/simple

其中 -i https://pypi.douban.com/simple 即切换镜像源下载。
安装好之后可通过 conda list 查看当前环境的库。
在这里插入图片描述

4. 安装PyCharm

目前使用较多的IDE是 PyCharm 或 VSCode ,个人觉得PyCharm比较专业一点,用起来也好用。
https://www.jetbrains.com/pycharm/download/#section=windows
到官网下载PyCharm,社区版是开源的,下载社区版,按照提示安装即可。
在这里插入图片描述

***** 使用方法 ******

打开PyCharm,新建或打开现有的项目。

Ctrl + Alt+S打开设置页,导入先前设置好的环境。

在这里插入图片描述
此处从Conda中选择现有的环境,并且勾选Make available to all projects,即应用到所有项目。
在这里插入图片描述

最后

至此,炼丹炉已修建完成。放入足够的原材料(数据集),使用合适的配方(模型算法),即可开始炼丹啦!
在这里插入图片描述

标签:RTX3060ti,版本,cuDNN,pytorch,CUDA,https,Win10,安装,下载
来源: https://blog.csdn.net/weixin_47345984/article/details/112265943

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有