Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

这篇具有很好参考价值的文章主要介绍了Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

1 配置 python 环境

1.1 安装 Anaconda

进入anaconda官网:https://www.anaconda.com/
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
点击 download 下载文件,我这里是 Anaconda3-2022.10-Windows-x86_64.exe(后续更新版本exe文件会有差别)

下载后打开 .exe 文件下载 anaconda:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

选择安装路径(用默认的路径也可以):

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

这里两个都选:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后安装就可以了。

1.2 检查环境安装成功

打开 cmd,输入 conda(如果是下图这样就说明 anaconda 安装成功了):
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
输入 python,这里可以查看 Python 的版本:
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

1.3 创建虚拟环境

在 cmd 中输入 conda create -n 环境名 python==版本号,如:conda create -n pytorch python==3.9.13(这里我自定义环境名为 pytorch,python 版本为 3.9.13)

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

输入 y

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

创建完就是这样的:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

1.4 进入/退出 刚刚创建的环境

进入环境:输入 conda activate 环境名,如 conda activate pytorch
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
退出环境:输入 conda deactivate

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

1.5 其它操作

1.5.1 查看电脑上所有已创建的环境

conda info --env
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

1.5.2 删除已创建的环境

conda remove -n 环境名 --all
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

2 安装 CUDA 和 CUDNN

2.1 查看自己电脑支持的 CUDA 版本

可参考:如何查看自己电脑当前版本CUDA 可兼容的最高版本

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

这就说明我的电脑可以安装的 CUDA 版本可以是 11.7.1 及以下的版本。

2.2 安装 CUDA

建议先去 pytorch 官网看下目前可以直接用指令安装的 CUDA 版本(主要是为了方便后续操作),此外,https://pytorch.org/get-started/previous-versions/ 也提供了以前版本的安装指令:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

看 Compute Platform,有 CUDA 11.6CUDA 11.7,而且我电脑支持的最高 CUDA 版本为 11.7.1,所以后续安装时就安 11.6 或 11.7 版本的 CUDA。

前往 CUDA 官网:https://developer.nvidia.com/cuda-toolkit-archive

这里以 11.7.0 版本为例:
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

按如下选择(Installer Type两种都可以,因为是国外网站,下载用外网,如果速度慢的话就选 exe(network)):
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

下载完后执行 .exe 文件。

根据自己需求选择是否更改安装路径:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

选择自定义:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

这个不用管,点下一步就行:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

选择安装位置(这里我改了路径,用默认的也可以,这个路径要记住,后续要用):

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

至此 CUDA 安装结束。


2.3 安装 CUDNN

进入官网:https://developer.nvidia.com/rdp/cudnn-download

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
先登录/注册账号,然后按如下选择:Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

将下载后的 zip 文件解压,里面的内容如下:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

找到刚才安装 CUDA 时选择的安装位置:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后将 CUDNN 中 bin 文件夹中的所有文件复制到 .../CUDA/v11.7/bin 中;
CUDNN 中 include 文件夹中的所有文件复制到 .../CUDA/v11.7/include 中;
CUDNN 中 lib/x64 文件夹中的所有文件复制到 .../CUDA/v11.7/lib/x64 中。

至此 CUDNN 安装结束。

2.4 检查 CUDA 安装成功(查看 GPU 使用率、显存占用情况)

在 cmd 中输入 nvidia-smi,同时这个指令也可以查看 GPU 的一些信息,如果出现如下界面就说明 CUDA 安装成功了,这是最好的(但好像有的电脑会报错 'nvidia-smi'不是内部或外部命令,这不一定就表示 CUDA 安装失败了,可以搜一下解决方案,或者暂时不用管,继续往后做,即使安装失败也不会影响后续的操作,后面安装完 pytorch 后还能用其他代码检查 CUDA 是否可用)。

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程


3 安装 PyTorch

3.1 安装 PyTorch

进入官网:https://pytorch.org/

点击 Install:
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
按如下选择(CUDA 版本要对应),下图中 Run this Command 中的指令后续要用到:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

打开 cmd,进入之前创建的 Python 环境,输入 conda activate 环境名,如:conda activate pytorch

进入环境后输入指令(来源于上图):conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia这里要挂下外网,不然安装很慢的,没有条件的话请移步 【3.3 其它方法】。注意,不要像一些教程那样用清华源,用清华源安装的 pytorch 没有 GPU 版本的,全是依靠 CPU 的,GPU 和 CPU 的算力差距很大,跑深度学习代码时用 CPU 能慢死

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
输入 y

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后等待安装就可以了,安装完是这样的:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

3.2 检查安装是否成功

可输入如下指令:

import torch
print(torch.version.cuda)  # 查看 CUDA 版本
print(torch.cuda.is_available())  # 查看 CUDA 是否可用(即训练时是否可用 GPU)
print(torch.cuda.device_count())  # 查看可行的 CUDA 数目

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

3.3 其它方法

如果没有加速器挂不了外网的话操作就比较麻烦了。这里再强调一遍!!!不要像一些教程那样用清华源,用清华源安装的 PyTorch 没有 GPU 版本的,全是依靠 CPU 的,跑深度学习代码时用 CPU 很慢。

在 PyTorch 官网中选择 Pip 的方法,找到如下网址:
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后我们进入打开这个网址,也就是 https://download.pytorch.org/whl/cu117

里面就是这样的:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

我们需要自己下载的文件就是 torchtorchvisiontorchaudio

然后我们先点击进入 torch,找到如下位置(根据自己的 CUDA 版本、python 版本找到相应的文件,cu117 指的是 CUDA 版本为 11.7,cp39 指的是 python 版本为 3.9,版本一定要对应上),点击下载:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

torchvisiontorchaudio 同理:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后我们把这三个 .whl 文件随便放到一个文件夹下:

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后打开 cmd,进入到这个文件夹,用 pip install 刚刚下载的文件 安装:

pip install torch-1.13.1+cu117-cp39-cp39-win_amd64.whl
pip install torchvision-0.14.1+cu117-cp39-cp39-win_amd64.whl
pip install torchaudio-0.13.1+cu117-cp39-cp39-win_amd64.whl
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

4 在 PyCharm 中使用 PyTorch

首先新建项目,按如下步骤操作:
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

然后点击 Create 创建项目即可

可通过以下代码测试使用 PyTorch:

import torch

print(torch.version.cuda)  # 查看 CUDA 版本
print(torch.cuda.is_available())  # 查看 CUDA 是否可用(即训练时是否可用 GPU)
print(torch.cuda.device_count())  # 查看可行的 CUDA 数目

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3, 3).cuda())

Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

5 远程 Linux 服务器配置 PyTorch

远程服务器配置 Anaconda 并安装 PyTorch 详细教程文章来源地址https://www.toymoban.com/news/detail-435764.html

到了这里,关于Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

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

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

相关文章

  • Orin 安装CUDA CUDNN TensorRT Opencv和输入法的环境配置

    有两种方法可以安装CUDA环境 在刷机完成的Orin,执行如下命令: 注释–如果在执行第三行命令,报错的话,先查看nvidia-l4t-apt-source.list 将其修改为 修改完后,重新执行上面那三行命令 CUDA 检查是否安装成功 运行命令 输出结果 如果报错,需要把nvcc添加到环境变量,方法如下

    2024年02月10日
    浏览(29)
  • 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日
    浏览(38)
  • Windows下Anaconda+CUDA+CUDNN+Pytorch+VSCODE安装配置及常见问题(可行方案)

    前言 此处写的以自己GT730为例,主要针对老版本显卡的可行安装步骤 这个内容主要是针对自己折腾几天遇到的问题和解决方法的一个记录,用来作为自己的备忘,遇到同样问题可以参考,同时也感谢各位博主之前的分享,为我提供了极大帮助; 另外,若有其它问题也可留言

    2024年02月04日
    浏览(39)
  • Jetson NX系统烧录以及CUDA、cudnn、pytorch等环境的安装

    这两步比较简单,所以略了。虚拟机的配置需要注意硬盘空间大一点,至少40G。 NVIDIA SDK Manager下载地址:https://developer.nvidia.com/drive/sdk-manager 直接执行会报缺少依赖的问题,然后执行命令: 安装依赖。等待依赖安装完成。 依赖安装完成后再次执行安装sdk的命令。 到这里sdk

    2024年02月03日
    浏览(33)
  • 【ubuntu环境配置】超详细ubuntu20.04/22.04安装nvidia驱动/CUDA/cudnn

    nvidia显卡驱动安装方式有三种:使用ubuntu附加驱动的方式;使用命令行方式安装;使用.run文件的方式进行安装, 点击菜单中的Additional Drivers选择适合的驱动版本进行安装,该方法最方便快捷(但有时会翻车) 更新所有的软件包 安装显卡驱动 详见我的另一篇博客Ubunut20.04/2

    2024年01月22日
    浏览(63)
  • 详解Windows系统下面如何查看CUDA、cuDNN、Python和各个软件包的版本

    这个题目网络上有很多的讲解,但是 查看CUDA、cuDNN版本 和 查看Python与自身各个软件包 是分开的,且cuDNN版本的查看方式似乎已经过时【截止2023-10-23】。由于自身需要且出于回馈 互联网大学 的目的,将相关内容重新整理在此篇博客中。 方法1:在Windows终端中通过以下3条下命

    2024年02月04日
    浏览(38)
  • 配置Cuda及Cudnn,在Anaconda创建虚拟环境,安装GPU版Pytorch,并在Jupyter noterbook及Pycharm中调用【极其详细】

    目录  一、配置Cuda及Cudnn (一)下载Cuda 1、查看电脑自带的Cuda版本 2、下载相应版本Cuda安装包 3、安装Cuda 4、配置变量 (二)下载Cudnn 二、创建虚拟环境 三、虚拟环境中安装GPU版Pytorch (一)有关环境的基本指令 (二)安装GPU版Pytorch 四、将虚拟环境在Jupyter noterbook及Pycha

    2024年04月28日
    浏览(35)
  • Windows电脑GPU配置CUDA和cuDNN

    前言:最近需要使用电脑的GPU进行并行计算,由于以前也配置过GPU的深度学习环境,但是并没有自己总结如何配置GPU的CUDA和CUDNN,因此趁着这次写了这篇文章,留作记录。 从未配置过 GPU 深度学习环境的话直接跳过 如果以前配置过对应的 GPU 深度学习环境,需要重新配置的话

    2024年02月09日
    浏览(40)
  • 安装CUDA以及CUDNN(windows版)

    笔者为了安装GPU版torch,所以需要先安装CUDA 以及CUDNN,本文介绍如何安装对应版本的CUDA和CUDNN,为了后面的深度学习做准备。 一、查看自己显卡驱动版本 打开NVIDIA显卡控制面板,点击系统信息,在这里查看驱动程序版本,比如我的本机是517.00.    根据自己主机版本打开官网

    2024年03月20日
    浏览(36)
  • 【CUDA】判断电脑CUDA和cuDNN是否安装成功(Windows)

    查看CUDA安装是否成功: 进入到CUDA的安装路径,找到如下两个.exe文件: 我自己安装完的路径为: C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1extrasdemo_suite 然后打开CMD窗口(以管理员身份): 先进入自己的目录: 首先执行: deviceQuery.exe ,查看是否出现如下界面: 然后执行

    2024年02月16日
    浏览(40)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包