Pytorch下载安装教程

Time：2024.05.05

进行此教程的前提，你已经安装完CUDA、cuDNN、PyCharm、Anaconda的下载安装

启动Anaconda Prompt(一般在开始菜单的所有应用里)，然后输入以下命令：

conda create -n pytorch python=3.9

使用conda创建一个名为pytorch的虚拟环境，并指定Python版本为3.9。

您可以根据自己的需求更改这个版本，但必须明确指定Python的具体版本。

创建成功后，输入以下命令查看安装的所有环境：

conda info --envs

激活虚拟环境，输入以下命令：

conda activate pytorch

打开pytorch官网：PyTorch，点击Get Started

根据安装CUDA版本选择pytorch的安装，最后复制Run this Command:后面的内容

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

进行到一半会弹出是否继续，输入y，回车

最后，最上行仅出现done

检验是否安装成功，输入python,回车

再输入

import torch

即安装成功。

再检验是否成功调用GPU,输入

print(torch.cuda.is_available())

返回Ture,即成功

END