PyTorch安装

PyTorch安装

安装前言

安装前言：PyTorch的安装需要根据自己电脑的配置对应，分为GPU和CPU版本，同时GPU版本需要与自己电脑的显卡版本号对应

有无独显	安装版本
有	GPU
无	CPU

PyTorch官网

PyTorch

查看电脑上是否有独显以及版本号

nvidia-smi

上图可以看出我的CUDA版本是12.0 ，如果没有显示就说明没有独立显卡，可以选择安装CPU版本

查看官网的安装界面

在刚刚的官网首页往下滑，可以找到安装界面

Stable是稳定版，Preview是预览版，这里建议选择稳定版；再根据自己的操作系统选择对应的，我这里选择的是Windows；Package是要使用的安装包，我这里选择Conda，因为我电脑上已经有了Anaconda（这个是一个python的版本和包管理工具，可以在电脑上创建不同的python环境，方便我们控制python的版本），如果你想安装Anaconda，可以参考我的一篇博客，里面有我当时学习安装的视频链接和教程Datawhale-ai夏令营-miniconda环境搭建-CSDN博客，如果你不想使用Conda命令安装，使用pip也可以进行安装。但是不管使用Conda或者Pip，都要确保你的Python版本要大于等于3.8。

通过这张图我们发现并没有CUDA12.0版本，但是翻阅社区了解到，cuda是向下兼容的，所以这里我们选择CUDA11.8版本，如果你安装CPU版本，直接选择CPU

Run this Command是下载的链接，这个要复制粘贴到对应python环境下的终端里，这个下载的包很大，然而它的服务器却在国外，所以如果你的python没有更换过镜像源，可以考虑修改安装的镜像源，否则可能要安装60分钟以上。

PyTorch安装

我的python环境：3.10.13

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

PyTorch测试

import torch
print(torch.cuda.is_available())

结果返回true就说明安装成功了

结语

在安装pytorch的时候可能比较慢，即使在换源的情况下，这个耐心等待就好，我的是安装了将近20分钟。