RTX3060安装pytorch

  • 1 安装anaconda
  • 2 查看本机显卡支持的cuda最高版本
    • (2)根据pytorch版本选择要安装的CUDA
    • (3)下载安装CUDA
    • (4)下载cudNN
    • (5)下载安装刚刚选择的pytorch版本

前不久刚刚入手了一台新电脑,显卡为RTX3060,在安装环境的时候,踩了不少坑,现在将经验总结如下:

1 安装anaconda

这个可以看这个教程:https://blog.csdn.net/in546/article/details/117400839
需要注意的是,要记得添加到环境变量中

2 查看本机显卡支持的cuda最高版本

在右下角中点击“NVIDIA设置”

在弹出的窗口中点击 帮助——系统信息

在弹出的窗口中,选择 组件 浏览页,并在下面找到 NVCUDA64.DLL,在产品名称那一栏可以看到本机支持的最高CUDA版本是11.2.162

如果是Linux系统,那么可以在命令行中输入nvidia-smi(我在一台服务器上演示这个命令,并非表示本机的RTX3060的显卡)

(2)根据pytorch版本选择要安装的CUDA

进入pytorch官网的下载页面:https://pytorch.org/get-started/locally/

可以看到,这里最新的稳定版本(1.10.1)目前仅支持CUDA 10.2和CUDA11.3两种CUDA环境,但我们的机器支持的最高版本是11.2,无法安装11.3的CUDA,而10.2的CUDA也无法安装,因为7.X算力的CUDA不能安装在8.X算力的显卡上,即算力不能跨。(加入我的显卡支持11.3的cuda,那就可以直接用命令了,后面的cuda和cudnn都可以不用装了,因为命令里面有一句cudatoolkit=11.3,这就是安装cuda工具的意思)

此时可以安装更早的pytorch版本,点击 install previous versions of PyTorch

在新跳转的网页中找到CUDA版本≤11.2对应的pytorch

我在v1.8.1下找到了支持CUDA11.1的pytroch版本

截图中的安装命令中,没有安装cuda的部分,因此接下来我们就安装cuda11.1
其实可以直接安装pytorch1.8.0,因为它的命令如下:

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

这里连同cuda也一起安装了,但我们这里还想演示一下cuda和cudnn的安装,所以就安装不带cuda的版本。

(3)下载安装CUDA

下载页面在这:https://developer.nvidia.com/cuda-toolkit-archive
我的这台机器,下面两个都可以


下载之后安装

(4)下载cudNN

根据cuda的版本下载对应的cudnn,下载页面在这:https://developer.nvidia.com/cuDnn
登录之后选择配套的版本下载

截图中的两个,哪个都行

下载并解压缩cudnn

将其中的所有文件和文件夹复制,然后在cuda的安装目录下粘贴,如果有提示,则替换。(CUDA默认安装在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA中)

(5)下载安装刚刚选择的pytorch版本

刚刚我们选择的是1.8.1版本的pytorch,我们在命令行键入

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

接下来就老老实实等待就行。

RTX3060安装pytorch相关推荐

  1. [学习笔记] ubuntu 下 anaconda 安装 pytorch

    搜索别人的资料大概罗列了几种方法在anaconda下安装pytorch,首先是在pytorch官网直接选择对应的pytorch版本,在终端输入直接安装 PICTURE:/home/liyifei/.c ...

  2. 利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装---免额外安装CUDA和cudnn(适合小白的保姆级教学)

    系列文章目录 ​​​​​​​深度学习原理-----线性回归+梯度下降法 深度学习原理-----逻辑回归算法 深度学习原理-----全连接神经网络​​​​​​​ 深度学习原理-----卷积神经网络 深度 ...

  3. 通过anaconda2安装python2.7和安装pytorch

    ①由于官网下载anaconda2太慢,最好去byrbt下载,然后安装就行 ②安装完anaconda2会自动安装了python2.7(如终端输入python即进入python模式) 但是可能没有设置环境 ...

  4. 记录一次简单、高效、无错误的linux上安装pytorch的过程

    1 准备miniconda Miniconda Miniconda 可以理解成Anaconda的免费.浓缩版.它非常小,只包含了conda.python以及它们依赖的一些包.我们可以根据我们的需要再安 ...

  5. PyTorch 笔记(01)— Ubuntu 使用 pip 清华源安装 PyTorch

    PyTorch 的安装可以到官网 https://pytorch.org/,选择适合自己机器以及安装方式,执行对应的命令即可. wohu@ubuntu1804:~$ sudo pip3 install ...

  6. 以安装PyTorch为例说明Anaconda在Windows/Linux上的使用

    在Windows10上配置完MXNet 1.3.0后,再配置PyTorch 1.0时,发现两者需要依赖的NumPy版本不一致,之前是通过pip安装NumPy,根据pip的版本不同,会安装不同版本的Nu ...

  7. Mac下pycharm如何安装pytorch

    通过虚拟环境virtualenv安装pytorch 首先提及,下面的命令,我都是直接在macos上的pycharm中的terminal执行的 1.安装virtualenv虚拟环境: pip insta ...

  8. 创建虚拟环境 安装pytorch 0.3.1

    (1)创建虚拟环境 https://blog.csdn.net/lyy14011305/article/details/59500819 (2)安装pytorch=0.3.1 conda instal ...

  9. 手把手教你用anaconda安装pytorch最新版

    序言: 在有过python和jupyter notebook 安装基础之上,安装pytorch. 小贴士:个人不建议用miniconda,虽然比较小巧,但是功能还是没办法和anaconda比较.下面开 ...

  10. pycharm安装pytorch报错 提示系列问题 torch 包找不到因为pip版本低,结果升级了pip从19.3到20.2 4又提示高版不支持torch安装

    pycharm安装pytorch报错 提示系列问题  torch 包找不到因为pip版本低,结果升级了pip从19.3到20.2 4又提示高版不支持torch安装 DEPRECATION: The - ...

最新文章

  1. oracle手工启动,SQLSERVER服务手工启动 批处理文件
  2. 药企巨头辉瑞与家用机器人 Mabu 合作,利用 AI 评估患者行为
  3. PHP MVC框架核心类
  4. silverlight中如何将string(字符串)写入Resource(资源)?
  5. gradle使用技巧之全局变量
  6. Redis中的过期策略
  7. 滑动listview隐藏和显示顶部布局
  8. codevs——2152 滑雪
  9. C# Random生成相同随机数的解决方案
  10. C言语教程第三章: C言语挨次妄想开端(7)
  11. Apple越狱后超级实用的Cydia常用源推荐
  12. lsass.exe是什么?
  13. 台式计算机调亮度快捷键,台式电脑怎么调亮度快捷键(显示器亮度的调节方法)...
  14. 前端布局layout总结,左右布局,上中下布局
  15. 武汉市查询社保电脑号及公积金账号的方法(湖北省其他市也适用)
  16. 利用matlab构建自己的colormap(色彩搭配)
  17. Ubuntu下解决firefox profile cannot be loaded it may be missing or inaccessible
  18. vmware workstation 未能启动vmware
  19. 微带贴片天线谐振边为什么是半波长?
  20. filezilla源码安装教程(含错误信息解决方案)

热门文章

  1. python计算均线斜率_一根20日均线闯天下——量化回测“压箱底指标”,值得读十遍!...
  2. Spring Boot整合Redis
  3. ps动作保存不覆盖原文件_Photoshop从入门到精通:图像的基本操作,新建打开保存关闭文件...
  4. 分时线的9代表什么_股权生命9条线,持有不同股权代表了什么?
  5. 为什么电脑桌面无计算机图标,我的电脑桌面图标都没名字了为什么
  6. #49:Photoshop技巧的大杂烩
  7. web前端学习(三)——HTML5的字体、特殊符号、插入图片及头部元素的相关标签设置
  8. WEB前端--Day9(动画)
  9. NSIS静默安装VC运行库插件
  10. 国产ADAS“再”突围