Pytorch安装与配置教程

释放双眼,带上耳机,听听看~!
本教程详细介绍了Pytorch、CUDA和cuDNN的安装与配置步骤,以及验证安装是否成功的方法。适用于想要在win系统上安装Pytorch的用户。

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第29天,pytorch以及cuDNN的版本支持情况。

Pytorch:pytorch.org/

cuDNN:developer.nvidia.com/rdp/cudnn-a…

Pytorch安装与配置教程

可以看到对于win系统,Pytorch支持的版本有10.2,11.3,11.6等。

Pytorch安装与配置教程

cuDNN支持的版本有10.2,11.5,11.4等等。

这里我推荐装10.2或者11.3,原因见开头,不多做赘述。这里以11.3为例。

3.下载CUDA

在CUDAhttps://developer.nvidia.com/cuda-toolkit-archive中,寻找CUDA Toolkit 11.3版本,然后寻找相应的版本下载即可。

Pytorch安装与配置教程

4.安装CUDA

安装CUDA,首先需要选择CUDA的临时解压路径,这个临时解压文件夹会在安装完成后自动删除,这里建议默认。

Pytorch安装与配置教程

解压完成后会有安装程序,同意即可。接下来的安装选项选择自定义:

Pytorch安装与配置教程

在安装CUDA中取消这个VS有关的组件:

Pytorch安装与配置教程

底下这三个也没必要,可安可不安,看个人选择:

Pytorch安装与配置教程

安装路径仍然建议默认,在Program Files中,方便以后寻找。建议记住这里的CUDA路径。

Pytorch安装与配置教程

然后一直确认最后关闭即可,这里不多赘述。

5.验证CUDA是否安装成功

我们在cmd中使用cd命令切换到刚刚CUDA的安装路径下的bin(二进制)文件夹下,再执行nvcc -V命令。

Pytorch安装与配置教程

可见CUDA是正确安装的。

cuDNN的安装

在cuDNN的版本中,选择支持该版本的CUDA即可,这里我们看到v8.5.0的cuDNN支持CUDA 11.X,说明兼容cuda11.x全系列。点击下载即可。

Pytorch安装与配置教程

接下来,解压该压缩包,然后复制其中的文件夹

Pytorch安装与配置教程

粘贴到CUDA的安装目录下,即完成了cuDNN的安装。

Pytorch安装与配置教程

验证是否安装成功

在cmd中进入到demo文件夹:路径为C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.3extrasdemo_suite

执行bandwidthTest.exe,如果运行结果出现了PASS即代表安装成功。

Pytorch安装与配置教程

再输入命令deviceQuery.exe查询设备。

Pytorch安装与配置教程

这里会显示你的GPU型号,以及PASS,表示CUDA和cuDNN都安装成功了。

Pytorch安装与配置教程

Pytorch安装

进入Pytorch官网pytorch.org/,选择需要安装的pyt…

Pytorch安装与配置教程

下载torch

Pytorch安装与配置教程

由于我们安装的torch版本为Stable1.12.1,因此我们需要查找前缀为:torch-1.12.1的文件。注意找的是GPU版本,cuxxx代表CUDA版本xxx。这里我们找到对应的cu113,然后点击下载。cp代表python版本,这里我们选择cp37版本的win下载。

Pytorch安装与配置教程

然后返回,进入torchvision。

下载torchvision

Pytorch安装与配置教程

torchvision的版本选择最新就好,但是要与cuda及python匹配,这里直接搜索cu113-cp37

Pytorch安装与配置教程

下载对应的版本即可。

新建pycharm创建项目,新建项目名称为PytorchTest, 文件名为HelloPytorch。输入以下代码测试

import torch

print("Hello Pytorch{}".format(torch.__version__))

结果报错,ModuleNotFoundError: No module named 'torch',因为当前环境没有torch。

我们可以用conda创建虚拟环境并安装torch。

首先再Terminal中输入conda create -n pytorch-gpu python=3.7,这里python版本与我们要安装的版本相同。

出现Proceed ([y]/n)? 直接输入y即可。

接着输入conda activate pytorch_gpu激活环境。

注意:进入conda虚拟环境后venv前面的提示会变成你的环境名称,如下:

Pytorch安装与配置教程

如果没有显示,则可能因为pycharm终端采用的是PowerShell。需要在设置中切换。

Pytorch安装与配置教程

换成如图所示的cmd终端即可。

Pytorch安装与配置教程

进入终端后切换到下载刚刚torch和torchvision的文件夹中

cd D:Developpytorch_install //因人而异,cd到你的下载torch和torchvision的文件夹中即可。

用pip安装torch

pip install "torch-1.12.1+cu113-cp37-cp37m-win_amd64.whl"

用pip安装torchvision

pip install "torchvision-0.13.1+cu113-cp37-cp37m-win_amd64.whl"

完成安装

Pytorch安装与配置教程

安装完成后,需要绑定该项目的解释器为这个虚拟环境。因此需要设置:

Pytorch安装与配置教程

找到该项目的Python解释器,然后点击齿轮,选择add:

Pytorch安装与配置教程

选择conda,找到已经存在的环境:

Pytorch安装与配置教程

查找你的anaconda安装的目录

这里可以搜索anaconda并打开文件位置。

Pytorch安装与配置教程

然后继续右键打开文件位置即可。

Pytorch安装与配置教程

打开后找到envs(环境)文件夹。

Pytorch安装与配置教程

找到刚刚创建的环境,复制文件路径到pycharm。

Pytorch安装与配置教程

在pycharm中选择该文件路径下的python.exe解释器即可

Pytorch安装与配置教程

然后一直ok,等待python解释器的重载即可,可能需要等一分钟。

然后重新尝试测试代码并运行。

import torch

print("Hello Pytorch{}".format(torch.__version__))

print(torch.cuda.is_available())

Pytorch安装与配置教程

返回true说明安装成功。

CUDA的卸载

首先,搜索控制面板并打开

Pytorch安装与配置教程

找到程序卸载

Pytorch安装与配置教程

可以在列表中找到有关于NVIDIA的相关组件,找到有关于CUDA的组件并卸载即可,其他的可以保留,因为高于该版本的CUDA会更新其他组件的。(本质上临近时间安装的都能卸载

Pytorch安装与配置教程

右键 -> 卸载 然后在卸载程序中卸载即可。剩下的方法类似。

Pytorch安装与配置教程

最后不放心的话可以用火绒等软件清理一下注册表。

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

图生图生成二维码方法-美女照片合成演示

2023-12-11 9:17:14

AI教程

深入了解Prompt Learning:预训练模型+Prompt学习

2023-12-11 9:30:14

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索