conda环境下-pytorch-2.0.0-GPU版本-镜像极速安装

        发现好多同学安装完cuda和cudnn之后不会安装pytorch，尤其是2.0版本，官方给的安装命令可能不适用。下面是我的安装过程。这里默认你已经装好cuda和cudnn，没有装的可以自行搜索教程安装，我的是cuda 12.1版本，显卡是GTX 1650。（建议直接上12.1版本，版本是可以向下兼容的，也就是可以装低版本的pytorch）

目录

1. 为pytorch创建独立的conda环镜

2. 升级pip

3. 安装pytorch

 4. 测试一下能用不

---

 

1. 为pytorch创建独立的conda环镜

        打开anaconda prompt，

        开始创建新环境，环境名为torch2 ，py版本是3.8。

conda create -n torch2 python=3.8

        创建成功后进入torch2环境：

conda activate torch2 

2. 升级pip

        后面早晚得升级。命令如下，俩里面自己挑一个能用的：

pip install --upgrade pip
python -m pip install --upgrade pip

3. 安装pytorch

        直接执行以下命令，速度嗖嗖的哈（记得关VPN）：

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

        安装完成之后输入命令：pip list，看看是否装好，下面是好的样子：

 4. 测试一下能用不

        依次输入以下命令：

python
import torch
print(torch.__version__)  # 查看torch版本号
print(torch.version.cuda)  # 编译当前版本的torch使用的cuda版本号
print(torch.cuda.is_available())  # 查看当前cuda是否可用于当前版本的Torch，如果输出True，则表示可用

        成功的样子：