PyTorch环境的配置（GPU）

 一、安装Anaconda（Windows版本）

在浏览器进入官网​​​​​ https://www.anaconda.com/ 下载anaconda后安装：

安装成功

二、创造虚拟环境

安装成功后，找到Anaconda Prompt并打开

这样就会进入base环境中，我们要在这里创建虚拟环境

输入 conda create -n Torch python=3.9  (其中Torch是虚拟环境的名字，3.9是python的版本，两者都可以自定义 )

输入之后等待创建成功

创建完毕后输入conda activate Torch激活Torch环境

可以看到左边的括号里从base变成了Torch，即成功创建并进入虚拟环境。

三、下载Torch

打开cmd终端输入nvidia-smi命令，查看cuda版本为12.8，此版本为最高可安装的torch包版本。

打开pytorch.org网站，获得下载链接（我这里选择的是12.6版本）

​

获得下载链接：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

输入到Torch虚拟环境中进行下载，等待下载完成

​

下载结束之后，检验是否成功：

输入python进入python环境后，输入

 import torch
print(torch.__version__)

若返回版本号2.7.1+cu126，则版本以正确下载。

输入print(cuda.is_available())

若返回True表示GPU可用。

torch下载成功。

四、将torch环境配置到pycharm

下载并安装pycharm(专业版)软件后，打开pycharm，在设置中找到“python解释器”——>选择左上角的“添加解释器”——>“添加本地解释器”——>“选择现有”——>类型为“Conda”——>环境为“Torch”,点击确定。

等待加载

加载完成之后右下角会出现我们创建的环境“Torch”

检验是否配置成功：

创建一个python文件，输入以下代码：

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

输出为True，则配置成功。

五：注意

一、在虚拟环境Torch中下载torch时，若使用国内镜像网下载，可能下载得到的torch不会带有“+cu126”，这就需要放弃使用国内镜像，而使用官网的命令进行下载则不会出现这种问题。

二、在pycharm中配置“Torch”环失败，可能是pycharm的问题，解决方法：

在设置中找到“终端”，选择Shell路径为cmd.exe。