【从零开始构建GPT模型】（三）Windows篇：Pytorch2.0安装教程

目录

一、确定conda安装命令

 二、Conda创建虚拟环境

2.1 查看虚拟环境

2.2 移除虚拟环境

2.3 激活虚拟环境

2.4 退出虚拟环境

2.5 安装包

三、在创建的虚拟环境上安装pytorch2.0

---

一、确定conda安装命令

首先，我们打开Pytorch2官网，Start Locally | PyTorch

找到如下信息，而后下拉，找到一个官方表格。

依据自己的信息，填写各表格项，Run this Command项目即是我们需要的安装命令。

如，我选择的Stable->Windows->Conda->CUDA12.1，对应的命令为：

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

其中，Windows下,Conda和CUDA安装教程，可参与本专栏前两篇。

http://t.csdnimg.cn/cKgPc

http://t.csdnimg.cn/sbBh8

 二、Conda创建虚拟环境

打开Anaconda Powershell Prompt，在开始菜单里。如果你的不在，可自行百度解决，或者按照我专栏文章重新安装。

conda创建虚拟环境的命令为：

conda create -n your_env_name python=x.x

以下会有些需要输入y的操作，自不必多说了。

（不建议）亦可使用指定为路径，在其他充裕空间的盘下创建，但此方法创建的虚拟环境无名称。在Anaconda安装路径下envs文件夹里的虚拟环境才配有姓名，因而，Anaconda的安装路径显得格外重要。

conda create --prefix=D:/Anaconda_CUDA/env_pytorch2/win_ai_pytorch2 python=3.11.5

1、your_env_name为自己确定的环境名称，这里我命名为：win_ai_pytorch2。

2、python=x.x，为python的版本（这里是anaconda安装时自带的python）。查看方法为：

在Anaconda Powershell Prompt中，输入即可。

python -V

 如此，我自己的conda虚拟环境创建命令为：默认创建路径为Anaconda安装路径下的envs。若不在此路径下创建指定路径的虚拟环境，默认会无名称，不建议使用。

conda create -n win_ai_pytorch2 python=3.11.5

在Anaconda Powershell Prompt中输入。

2.1 查看虚拟环境

在Anaconda Powershell Prompt中，输入

conda env list

2.2 移除虚拟环境

如果你是默认路径创建的虚拟环境，在Anaconda Powershell Prompt中，输入

conda remove -n your_env_name --all

如果你是指定路径创建的虚拟环境，请输入，-p后边为你自己的路径，这里是我的例子。

conda env remove -p D:\Anaconda_CUDA\env_pytorch2\win_ai_pytorch2

2.3 激活虚拟环境

在Anaconda Powershell Prompt中，输入

conda activate your_env_name

 如果是指定路径，请加上指定路径。

2.4 退出虚拟环境

在Anaconda Powershell Prompt中，输入

conda deactivate

2.5 安装包

在Anaconda Powershell Prompt中，输入

conda install -n your_env_name [package]

三、在创建的虚拟环境上安装pytorch2.0

创建完虚拟环境后，查看虚拟环境如下。

激活创建的虚拟环境

输入安装命令

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

耐心等待，结果如下

依次输入

python

import torch

torch.cuda.is_available()

 结果为True。

这样就可以了。

下次使用pytorch2.0，别忘了在哪个虚拟环境中安装的。