简单安装pytorch环境

1.升级pip命令

python -m pip install --upgrade pip

2.将全局的包索引地址设置为清华大学镜像源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

3.创建一个3.9.0版本的python虚拟环境，虚拟环境名自行设置，将env_name修改即可

conda create -n env_name python=3.9.0 -y

4.进入虚拟环境

activate env_name

5.pip 安装CPU版本的torch

pip3 install torch torchvision torchaudio

6.检查pytorch环境是否安装成功，在安装好的环境里输入python，然后输入以下代码

import torch
print(torch.__version__)

如果成功出现版本号说明安装成功

安装GPU版本

1.安装cuda和cudnn

（1）首先查看自己的电脑是否能够安装cuda和cudnn，以及能够安装的版；Win+R打开运行窗口，输入以下命令

nvidia-smi

或者通过搜索nvidia ，进去之后点击左下角的系统信息。然后选择组件，就可以查到主机允许安装的最高版本的cuda。

如图可知本机可支持安装的最高的cuda版本是11.6 .

2.接下来最好是创建一个虚拟环境，进行环境管理。在安装了anaconda的情况下，使用win+r，通过cmd打开命令，输入：

conda create -n pytorch python=3.9.0 -y

 创建一个名为：pytorch的虚拟环境：

接着输入：

conda activate pytorch

进入虚拟环境后；
先进pytorch官网：

PyTorch

然后可以选择安装以前的版本，找到适合自己电脑的cuda版本，或者低于自己电脑版本的cuda，例如我的是cuda11.6，就选择下面这句：

pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116

在命令行进行下载即可。

3.检验是否安装好

在命令行下输入python，进入编译：

然后输入下面的代码：

import torch
torch.cuda.is_available()

如果返回是Ture，就说明安装成功了。