【Pytorch】安装保姆级教学

最新推荐文章于 2024-12-26 17:13:27 发布

Anning9455

最新推荐文章于 2024-12-26 17:13:27 发布

阅读量589

点赞数 1

文章标签： pytorch python 人工智能

本文链接：https://blog.csdn.net/m0_48643017/article/details/129197370

版权

本文介绍了如何在Windows环境下，使用Anaconda进行PyTorch的离线安装，包括检查CUDA版本，从CUDA官网下载并安装，然后找到匹配的PyTorch和torchvision的离线whl文件进行安装。安装完成后，通过`nvcc-V`和代码验证安装是否成功。

摘要由CSDN通过智能技术生成

Pytorch简介

pyTorch是一个Python的开源机器学习库。它用于自然语言处理等应用程序。它最初由Facebook人工智能研究小组开发，而优步的Pyro软件则用于概率编程。最初，PyTorch由Hugh Perkins开发，作为基于Torch框架的LusJIT的Python包装器。有两种PyTorch变种。
PyTorch在Python中重新设计和实现Torch，同时为后端代码共享相同的核心C库。PyTorch开发人员调整了这个后端代码，以便有效地运行Python。他们还保留了基于GPU的硬件加速以及基于Lua的Torch的可扩展性功能。

Pytorch[GPU版本]环境部署

因在线pip和anaconda安装问题层出不穷(也可能是因本人能力原因)，本文采用离线安装的方式进行讲解。
环境：pycharm、anaconda[python3.9版本]、windows
本文为采用创建虚拟环境的方式进行操作，而是直接在anaconda环境下安装torch
离线whl网站：https://download.pytorch.org/whl/

在这里插入图片描述

1、cudnn

首先cmd命令行窗口，输入

nvidia-smi

在这里插入图片描述
本处我电脑是CUDA版本11.1，下载安装的torch匹配的cuda版本应在电脑本机CUDA版本及其以下。（本处我电脑安装cuda版本最高支持到11.1）
CUDA官网下载链接，下载速度可能略慢，可能需要科学上网
CUDA下载官网
本处版本号参考前两位即可，第三位为小更新，不影响总体安装环境配置。
在这里插入图片描述
2. 下载完时候双击打开，进行安装，记住不要勾选visual code

在这里插入图片描述

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

在这里插入图片描述

位置不懂的话，默认即可，如果改动，需要记住位置，本人直接默认了。

在这里插入图片描述
到此就安装完成了，如果测试自己安装是否成功，在cmd中输入

nvcc -V

在这里插入图片描述
显示版本号成功！

2、torch

本次采取是离线安装Torch，需要对应好torch和torchvision版本以及操作系统（windows64、32 、Linux等等），
安装之前先去网站查询一下，以防没有对应的离线安装包离线安装包网站

1首先点击torch，找到你的cuda版本和操作系统版本对应的torch版本，以及python版本。（此处对应错误，后面无法正常启用GPU）
在这里插入图片描述
cu是你下载的cuda版本，101即为10.1，cp是python版本35就是3.5，win_arm64就是64位windows系统，
torch之后的1.5.1就是你下载的torch版本，

3、torchvison

同理，根据你下载的进入torchvision下载对应的版本，torch和torchvison缺一不可。
在这里插入图片描述

安装

进入anaconda prompt
cd进入需要安装的python目录下

使用 pip install 拖入路径回车安装即可，安装torch和torchvision，安装两个，本操作不做演示。

验证是否安装成功代码：

 import torch
 
print(torch.__version__) # 这个输出torch版本
 
print(torch.cuda.is_available()) # 这个输出结果必须为True