四、Windows下安装pytorch_conda install pytorch==2.2.0 torchvision==0.17.0 t

目录

前言

1. 创建虚拟环境并激活

2.找pytorch的安装命令

3. 安装pytorch

4. 测试pytorch是否安装成功

5. 在Pycharm中打开项目文件，并配置pytorch环境

---

前言

我安装的是 python 3.11.0    pytorch 2.2.0     CUDA 12.1

python和pytorch是我不想装最新的，安装什么版本看自己了

CUDA版本要看电脑显卡的配置，如何选择CUDA版本可以看我另一篇文章：

三、Windows下，安装显卡驱动最新版本，确定要安装的pytorch的cuda版本-CSDN博客

1. 创建虚拟环境并激活

打开Anaconda Prompt命令行，刚打开时，位于base虚拟环境里，前面是(base)

我是创建一个 名为wuyu，python版本为3.11.0 的虚拟环境。虚拟环境名自己起。

输入下行命令后回车，遇到Proceed([y]/n)?后输入y，回车。

conda create -n wuyu python=3.11.0

然后激活上面创建好的虚拟环境

conda activate wuyu

激活后前面变换成(wuyu)

2.找pytorch的安装命令

pytorch官网：https://pytorch.org/

pytorch历史版本网址：https://pytorch.org/get-started/previous-versions/

我选历史版本中的2.2.0的版本，CUDA12.1，使用的是下行命令

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

3. 安装pytorch

在激活好的虚拟环境力里输入pytorch安装命令后回车，遇到Proceed([y]/n)?后输入y，回车。

安装成功后会清空命令行界面，然后显示done

下面这行命令是显示当前虚拟环境下安装的所有包

conda list

输入这行命令回车，可以看到安装上了python和pytorch

4. 测试pytorch是否安装成功

输入python回车
出现>>>, 输入import torch回车
安装CPU版本的pytorch的人，到此，不报错，说明pytorch安装成功
输入torch.cuda.is_available()回车
安装CUDA版本的pytorch的人，显示Ture，说明pytorch安装成功，显示false，说明没安装成功

5. 在Pycharm中打开项目文件，并配置pytorch环境

 双击打开Pycharm，

找到要打开的项目文件，我这里以mmdetection-3.3.0这个项目文件夹为例

信任此项目

它说为项目文件创建虚拟环境，点击取消，（我们不在此创建环境，我们在anaconda prompt命令行中创建，然后配置）

打开项目文件夹后，为项目配置python解释器，也就是带pytorch的虚拟环境

这样就可以用pytorch了。