如何在anaconda虚拟环境下安装pytorch【简单易上手版】

1.先创建conda的虚拟环境

conda create -n gputorch python==3.10

2.激活刚刚创建好的虚拟环境

conda activate gputorch

3.设置镜像源，这一步是后面安装pytorch相关包所需要的来源

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

4.输入以下命令，采用pip命令安装，这个和conda安装只是来源不同罢了，实现效果一样的，不会出现像采用conda指令安装那么多麻烦。

pip install torch==2.1.2+cu121 torchvision==0.16.2+cu121 torchaudio==2.1.2+cu121 -f https://download.pytorch.org/whl/torch_stable.html

【注】你这里查看一下，你自己的显卡驱动所能支持的版本，我这里（显卡4060）显卡驱动支持cuda12.3，所以我安装的cuda和pytorch都很新，你只需要根据自己需要更改上面的版本数字就好了。

例如：安装这个CUDA11.8版本的，这里还是建议版本后面的数字都是一致的比较好，如torch==2.1.2对应torchvision==0.16.2和torchaudio==2.1.2，后面都是.2， 如果torch是后面.1版本，另外两个也选择.1版本。

pip install torch==2.0.1+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html

5.检查是否安装成功

import torch 

torch.cuda.is_available()

若返回ture就是安装成功