目录

  • 1 前言
  • 2 CUDA&cuDNN
    • 2.1 查看硬件
    • 2.2 安装
    • 2.3 验证
  • 3 PyTorch
    • 3.1 安装
    • 3.2 验证
  • 4 总结

1 前言

本笔记仅作自学和记录用途。
在这里附上cudnn11.3的网盘资源

「cuda」https://www.aliyundrive.com/s/7PoVqL1tVrX 提取码: mi99
点击链接保存,或者复制本段内容,打开「阿里云盘」APP ,无需下载极速在线查看,视频原画倍速播放。

PyTorch和CUDA的版本配对 是新手最容易出错的点。建议提前先去Pytorch官网查看目前支持的pytorch和相应的CUDA版本。
看图可知目前最稳定的PyTorch版本是1.10.2,可支持的CUDA版本是10.2和11.3。我根据显卡版本选择的是CUDA11.3及PyTorch1.10。

2 CUDA&cuDNN

参考链接
深度学习环境搭建(GPU)CUDA安装(完全版)

2.1 查看硬件

查看显卡型号

2.2 安装

CUDA toolkit安装官网下载需要的版本,接下来的安装步骤同参考链接。

cuDNN安装官网下载需要的版本,接下来的安装步骤同参考链接。

p.s. 如果是CUDA11.3,就选择下面这个版本的cuDNN,安装好后会自动变成11.3版本以适配CUDA版本。
p.s. 安装地址一定要对应,环境变量也要设置好。

2.3 验证

打开cmd命令框,输入 nvcc -V,可以看到CUDA版本即可。


如图所示是CUDA11.3版本。

3 PyTorch

3.1 安装

参考链接
pytorch,torch,torchvision的gpu版本安装避坑

建议离线安装。在线安装就算使用镜像网站还是下载缓慢且成功率低。
下载torch和torchvision网址:http://download.pytorch.org/whl/torch_stable.html
下载需要的版本。
下载完成后,在conda命令框中进入要安装的环境,使用

pip install 该文件地址

命令安装torch和torchvision文件即可。

3.2 验证

进入conda命令框的环境,输入python验证是否装好。

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

查看输出结果是否正确即可。

4 总结

第一次安装CUDA的时候,在按照别的帖子的过程中没有注意安装地址导致一些文件的位置不正确。同时安装的是CUDA11.6版本,没有支持的PyTorch版本,因此在PyTorch安装好后,测试GPU是否运行得到FALSE的结果。因此在安装环境的时候一定要注意不同软件互相支持的版本以及安装位置的选择。

CUDA11.3以及PyTorch-GPU版本安装相关推荐

  1. pytorch GPU版本安装 win10

    我原先装的是pytorch cpu版本,在复现论文源码的时候提示AssertionError: Torch not compiled with CUDA enabled,查了一下资料,又看了一下电脑的 ...

  2. pytorch gpu版本安装

    1,安装anaconda 2,查看显卡版本信息 3,安装对应的驱动程序cuda tooltik 4,安装pytorch 首先安装anaconda ,安装anaconda的目的在于创建虚拟环境,因为有的 ...

  3. PyTorch 学习by_LX(一):pytorch GPU版本安装等碎碎念

    新买的RTX2070到了,一顿操作后装的是cuda11.0,而pytorch目前支持的最高cuda版本为10.2:无奈需要重装,本想在不卸载cuda11.0的条件下安装10.2,一顿操作猛如虎,结果运 ...

  4. 关于pytorch GPU版本安装的问题

    在官网使用 conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia 命令安装 结果是CPU ...

  5. pytorch,torch,torchvision的gpu版本安装避坑

    pytorch,torch,torchvision的gpu版本安装避坑 在线or离线安装 pytorch torch和torchvision 在线or离线安装 最近使用pytorch跑深度学习模型,需 ...

  6. Pytorch GPU版本whl文件安装

    Pytorch GPU版本whl文件安装 安装pytorch的时候,用pip安装时网速实在太慢,换源也不太行,1.2G的文件,一个网络波动就开始疯狂红字.因此使用whl文件进行安装 用whl文件进行安 ...

  7. Pytorch:入门指南和 PyTorch 的 GPU版本安装(非常详细)

    Pytorch: 入门指南和 PyTorch 的 GPU版本安装(非常详细) Copyright: Jingmin Wei, Pattern Recognition and Intelligent S ...

  8. 【深度学习】PyTorch 历史版本安装-祖传老代码运行刚需

    最新 PyTorch 安装 以及 CUDA 版本 如果要安装最新的 PyTorch 其实是很简单的,直接到官网首页就有各种系统的安装方法,如同所示: 这里有一个是 CUDA 版本,实测需要一模一样,这 ...

  9. TensorFlow GPU 版本安装个人总结:Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0

    TensorFlow GPU 版本安装个人总结:Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0 接触机器学 ...

  10. GPU版本安装Pytorch教程最新方法

    目录 步骤 第一步:安装 Anaconda 和 Pycharm 软件 第二步:下载安装CUDA11.3 (1)首先查看自己电脑GPU版本 方式一:搜索框输入nvidia,打开nvidia控制面板 方式 ...

最新文章

  1. input不管用 vue_Vue自定义指令实现快速读取Excel
  2. git命令从入门到实战,看这一篇足矣
  3. 【Visual C++】游戏开发笔记四十一 浅墨DirectX教程之九 为三维世界添彩:纹理映射技术(一)...
  4. 二十三种设计模式-六大原则
  5. tools URL 收集
  6. Druid的外网访问异常
  7. 数独游戏技巧从入门到精通_中国茶艺技巧:500集从入门到精通教程,视频+素材+笔记...
  8. 计算机视觉基本原理介绍—RANSAC
  9. Dom 和 Jquery,json对象的使用方法
  10. HW浮动静态路由及负载均衡
  11. Java中字符串的全部知识_java基础教程之字符串的介绍,比较重要的一个知识点「中」...
  12. php在屏幕中间弹窗,屏幕中间弹框的一种写法
  13. linux设置ipsan_Linux挂载IPSAN和FCSAN操作
  14. wps excel 操作指南
  15. 数字签名的原理和应用
  16. 跟着迪哥学python 经管之家_跟着迪哥学:Python数据分析与机器学习实战
  17. folly::ConcurrentSkipList 详解
  18. 如何完成微信小程序的登录功能
  19. 反射率(Reflectance)与反照率(Albedo)
  20. linux同步webdav,Linux系统使用WebDAV自动挂载私有云盘

热门文章

  1. 微信直连支付通道刷脸支付用户开通步骤
  2. oracle如何设置连接数,关于Oracle连接数设置
  3. OpenSIPS实战(八):修改sip消息-使用lumps system
  4. Go语言创始人从Google离职
  5. python+minicap的使用 安卓手机投票到电脑
  6. js仿苹果风格弹出框alert插件
  7. JAVAWEB开发之——Linux命令实战、虚拟机和centos的安装以及联网、常用命令、vi编辑以及软件的安装、sftp的使用
  8. 单机100万连接,每秒10万次请求服务端的设计与实现(三) - 变量共享、超线程与高性能队列
  9. 六方ZnS包裹碳纳米管复合粉体/PbS纳米晶原位包裹碳纳米管的复合材料/壳聚糖包裹的多壁碳纳米管
  10. C语言调用jni中JNIEnv指针使用和理解