pytorch的安装(CUDA10.2+cuDNN8.3.3+torch1.10+​torchvision​0.11.1+python3.9)

这篇具有很好参考价值的文章主要介绍了pytorch的安装(CUDA10.2+cuDNN8.3.3+torch1.10+​torchvision​0.11.1+python3.9)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

(已存网盘和硬盘,文件夹含三个文件)

本文基本逻辑是:

一、先根据电脑硬件的条件获取本身CUDA版本,据此以及表格比较得出cuDNN、torch、torchvision、python版本。

二、在NVIDIA官网下载CUDA和cuDNN,获取torch的下载链接,网页提供python3.9的下载链接

三、安装CUDA后,把cuDNN这个补丁装到CUDA里边,因为CUDA toolkit是CUDA的工具包,cuDNN是CUDA的加速器补丁;用环境为python3.9的pycharm执行官网的pip命令操作完成torch和torchvision的下载

最后检查

————————————————————————————————————————

这是最开始需要下载的东西(不下也行,反正本文需要python3.9以及pycharm)

python+anaconda+pycharm的安装,其中python3.11.1、conda 22.9.0

但是本文需要python3.9就很寄,不过没事可以下载两个python解释器

目录

一、确定版本部分

1.查看NVIDIA CUDA显卡的版本:10.2

2.查看CUDA(10.2)与cudnn(选择8.3.3)、PyTorch(选择1.10)、PyTorch(1.10)与torchvision(选择0.11.1,需要python3.9不能3.11)对应关系

二、下载部分

1. 下载CUDA Toolkit10.2

 2.下载cuDNN8.3.3

3.下载torch1.10和torchvision0.11.1

三、安装部分

1.CUDA Toolkit安装

  2.cuDNN8.3.3安装配置

3.torch 和 torchvision 的安装

检查部分


一、确定版本部分

windows10 版本安装 CUDA ,需要下载两个安装包CUDA toolkit(toolkit就是指CUDA工具包,CUDAToolkit与cuda版本相同)和cuDNN(用于深度神经网络的GPU加速库),两者都需要安装才可以调用GPU加速

总结部分:

CUDA version 10.2、CUDAToolkit10.2(电脑固定)

pytorch1.10 、cudnn8.3.3、torchvision 0.11.1、python 3.9(选择)

2023/5/8补:本人后来换了个电脑cuda是12.1版本,安装的对应版本的是:

CUDA version 11.6

cudnn8.4.0、 python 3.9 

Torch:1.7.1  Torchvision:0.8.2(安装命令:

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

1.查看NVIDIA CUDA显卡的版本:10.2

操作:cmd->nvidia-smi

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

  Driver Version: 430.90       CUDA Version: 10.2意思是:

显卡驱动版本 430.90       显卡支持最高CUDA版本10.2,意思是CUDA版本小于等于10.2的都可以安装上。

【2023/5/8补:换了个电脑,CUDA版本是12.1,但是在官网上是没有这个对应版本的,不必惊慌,下载11.6版本就可以,下图是验证版本

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 

2.查看CUDA(10.2)与cudnn(选择8.3.3)、Torch(选择1.10)、Torch(1.10)与torchvision(选择0.11.1,需要python3.9不能3.11)对应关系

cuda与cudnn对应的版本
cuda  : cudnn
11.5   :8.3.3、8.3.2、8.3.1、8.3.0、8.2.1、8.2.0
11.4   : 8.2.4、8.2.2、8.2.1、8.2.0
11.3   : 8.2.1、8.2.0
11.2   :8.2.1、8.2.0、8.1.1、8.1.0
11.1   : 8.2.1、8.2.0、8.1.1、8.1.0、8.0.5、8.0.5、8.0.4
11.0   : 8.2.1、8.2.0、8.1.1、8.1.0、8.0.5、8.0.4、8.0.3、8.0.2、8.0.1
10.2   :  8.3.3、8.3.2、8.3.1、8.3.0、8.2.4、8.2.2、8.2.1、8.2.0、8.1.1、8.1.0、8.0.5、8.0.4、8.0.3、8.0.2、8.0.1、7.6.5
10.1  : 8.0.5、8.0.4、8.0.3、8.0.2、7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0
10.0  :  7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0、7.4.2、7.4.1、7.3.1、7.3.0
9.2    :  7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0、7.4.2、7.4.1、7.3.1、7.2.1、7.1.4、7.1.2

Torch与CUDA 对应的版本

Torch 版本 CUDA 环境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 9.2
1.2.0、1.1.0、1.0.0(1) 10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1) 10.2
1.7.0(1) 11.0
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0 11.1
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1) 11.3
1.11.0 11.5
1.12.0(1) 11.6

torch 对应的torchvision和python版本

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

二、下载部分

CUDAToolkit10.2(电脑固定)

cudnn8.3.3、torch1.10 、torchvision 0.11.1、python3.9(根据对应情况选择)

1. 下载CUDA Toolkit10.2

CUDA Toolkit官方下载地址

找到对应版本10.2

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习这个就是下载之后的安装包

 2.下载cuDNN8.3.3

https://developer.nvidia.com/rdp/cudnn-archive

往下滑,找到对应版本

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 双击之后,跳出下载界面,或者直接下载这个:https://developer.nvidia.com/compute/cudnn/secure/8.3.3/local_installers/10.2/cudnn-windows-x86_64-8.3.3.40_cuda10.2-archive.zip

最后得到:cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

3.下载torch1.10和torchvision0.11.1

这个链接是所有torch版本https://pytorch.org/get-started/previous-versions/

打开后下滑

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 这个是conda下载,不用这个

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

这个是pip下载,用pip这样子的。下滑找到对应的版本,

我的是win10系统+cuda10.2+pytorch1.10,所以就是下图蓝色部分:

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

三、安装部分

1.CUDA Toolkit安装

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

双击运行-自定义,不过要注意的是

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

如果你是第一次安装,尽量全选
如果你是第n次安装,尽量只选择第一个,不然会出现错误

(总之新安装就是全默认)

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

下载完成后,查看系统变量中是否添加了路径,如果没有需要自己添加

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

测试环境是否安装成功: cmd打开命令提示符,输入nvcc -V、set cuda得到有效输出则按安装成功。

 cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

————————————CUDA Toolkit10.2安装完毕————————————

  2.cuDNN8.3.3安装配置

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

解压这个文件,得到下图

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 把三个文件夹拷贝到cuda的安装目录下:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 让提供管理员权限就提供就行,cuDNN 其实就是 CUDA 的一个补丁而已

验证是否成功:

网上都说需要在系统环境变量中的 path 添加如下路径

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp

但是我没添加,测试也成功了。。。

测试方法是,在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite路径下,cmd,然后输入     .\bandwidthTest.exe   

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

result=PASS就是ok 

————————————cuDNN8.3.3配置完毕————————————

3.torch 和 torchvision 的安装

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 复制好 ,然后打开pycharm,这里是3.11版本的python环境

1.点开终端,2.粘贴安装代码cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 出现错误:

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

问题是没有3.11版本对应的torch,所以下载python3.9,把pycharm的环境改成3.9再操作:

就得是python3.9环境下,点开pycharm后,在下边的“Terminal”点一下,然后粘贴上蓝色的部分

pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

如下图:

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 最后等待几分钟:

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 验证是否成功:在pycharm里建一个py文件执行下面代码

import torch
if __name__ == '__main__':
    print(torch.__version__)
    print(torch.cuda.is_available())
    print(torch.version.cuda)

 返回:

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

 结束!

————————————————————————————————————

检查部分

cuda10.2对应的pytorch,安装与卸载,深度学习,pytorch,python,深度学习

可以用这6个语句检查一下 文章来源地址https://www.toymoban.com/news/detail-796672.html

set cuda
nvcc -V
PYTHON
import torch
print(torch.__version__)
print(torch.version.cuda)

到了这里,关于pytorch的安装(CUDA10.2+cuDNN8.3.3+torch1.10+​torchvision​0.11.1+python3.9)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • win下pytorch安装—cuda11.6 + cudnn8.4 + pytorch1.12 + tensorRT(pycuda)

    写在前面 博主这里装的是cuda11.7,最后一步tensorRT运行的时候有个pycuda的安装,它的最新版本只支持到cuda11.6,所以博主最后是又把cuda11.7卸载后重新安装了11.6,安装过程和11.7一样。pytorch对应的版本也应该修改,但过程都一样。 下载地址:cuda下载官网链接 这里有个前置工作

    2024年02月11日
    浏览(50)
  • 【Ubuntu20.04 CUDA11.1+Torch1.10+Anaconda 保姆级安装教程】

    安装CUDA时需要和Torch版本对应起来,最好先去torch官网上确定要安装的torch版本对应的CUDA版本。 在安装CUDA之前需要先确定是否已经安装驱动,打开终端输入 nvidia-smi ,若有输出,则表明驱动安装过,否则需要先安装驱动(驱动安装教程) 1、CUDA11.1下载 先去CUDA官网上下载要安

    2024年02月02日
    浏览(66)
  • cuda10.2安装python3.7pytorch

    1.conda创建虚拟环境 2.激活虚拟环境 3.安装 pytorch 去 pytorch 官网找对应的版本: https://pytorch.org/get-started/previous-versions/ 酷酷下载······

    2024年02月01日
    浏览(43)
  • Ubuntu22.04 LTS + CUDA12.3 + CUDNN8.9.7 + PyTorch2.1.1

    本文记录Ubuntu22.04长期支持版系统下的CUDA驱动和cuDNN神经网络加速库的安装,并安装PyTorch2.1.1来测试是否安装成功。 如果是旧的不支持UEFI启动的主板,请参考本人博客U盘系统盘制作与系统安装(详细图解) 如果是新的支持UEFI启动的主板,请参考本人博客UEFI下Windows10和Ubun

    2024年02月03日
    浏览(43)
  • 【pytorch】torch1.7.1安装、查看torch版本、GPU是否可用

    在conda 虚拟环境下安装torch==1.7.1+GPU版本 本机环境 CUDA 11.0 Python 3.7 安装torch1.7.1 官网搜索确认需要下载的对应本机cuda的torch版本,使用在线下载即可,会直接安装好torch、torhvision、torchaudio。 1、官网搜索对应cuda的版本 2、安装命令 查看安装版本

    2024年02月12日
    浏览(64)
  • ubuntu20.04系统4060安装cuda11.8和cudnn8.6

    在终端输入: 安装代码: 重启电脑(这一步很重要,不重启没有效果): 安装命令 CUDA默认安装在/usr/local/目录下,一般Change Toolkit Install Path可以不做修改。但如果是普通用户安装,需设定安装路径为用户主目录下,光标移动到Change Toolkit Install Path按回车,手动修改安装路径后按

    2024年01月16日
    浏览(60)
  • windows10操作系统 显卡MX150 安装CUDA+cuDNN+pytorch

    本人水平有限,如有问题请多指正 笔记本系统:Windows 10 64位 显卡:NVIDIA GeForce MX150 显卡驱动程序版本:512.78 显卡驱动程序支持CUDA版本: 11.6.134及以下 安装CUDA:11.6.0 安装cuDNN:8.7.0 一、安装CUDA 1、确定CUDA版本 查看本机驱动程序版本。打开“NVIDIA 控制面板”,点击“帮助”

    2023年04月10日
    浏览(62)
  • yolov7的 TensorRT c++推理,win10, cuda11.4.3 ,cudnn8.2.4.15,tensorrt8.2.1.8。

    yolov7的 tensorrt8 推理, c++ 版本 环境 win10 vs2019 opencv4.5.5 cuda_11.4.3_472.50_win10.exe cudnn-11.4-windows-x64-v8.2.4.15 TensorRT-8.2.1.8.Windows10.x86_64.cuda-11.4.cudnn8.2.zip RTX2060推理yolov7, FP32 耗时 28ms  ,FP16 耗时 8ms,单帧对应总耗时30ms 和 10ms。 推理yolov7-tiny,FP32 耗时 8ms  ,FP16 耗时 2ms。 tensorr

    2024年02月15日
    浏览(45)
  • CUDA11.1、cuDNN8.6.0、Tensorrt8.5.3,ubuntu20.04安装过程记录

    CUD11.1 下载地址:CUDA Toolkit Archive | NVIDIA Developer  安装: 对于不是sudo用户,可以不执行sudo,不过没办法装到/usr/local/,可以装到你有权限的文件夹目录。 装完后,需要增加环境路径到~/.bashrc并source  记得source ~/.bashrc使得环境生效 cuDNN8.6.0,这个版本我是根据我需要安装的T

    2024年02月15日
    浏览(48)
  • Ubuntu20.04下载cuda11.3+cuDNN8.2.0,卸载cuda11.6+cuDNN8.4全记录【保姆级教程】

    CUDA(Compute Unified Device Architecture)是由NVIDIA公司推出的一种高性能并行计算架构。它利用GPU的并行处理能力,能够显著提高计算效率,尤其在科学计算、数据分析、深度学习等领域具有广泛应用。CUDA提供了一套编程模型和API,使开发人员能够利用GPU并行执行计算任务,从而获

    2024年02月05日
    浏览(96)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包