一、首先安装好 Anaconda
(此处不详细介绍,如果有没有安装Anaconda的,可以看一下下面我发的清华镜像网站,自行下载,注意:不建议下载最新版本)
Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
二、安装 pytorch
1.先打开 anaconda prompt
2.创建虚拟环境
conda create -n 虚拟环境名字 python=你要下载的版本以我的为例:
2.1创建虚拟环境,我的是 python3.9
conda create -n pytorch390 python=3.92.2 输入 y
2.3 安装完成后,输入 conda env list 检验是否安装成功。
conda env list3.进入刚才我们已经创建好的虚拟环境
输入:
conda activate pytorch3904.查看CUDA版本
输入:
nvidia-smi因为我的CUDA是12.4版本,所以等下我只需要下载一个比12.4版本低的就可以了。
5.搜索 pytorch官网 或者 点解我下面我的链接也可以
PyTorchhttps://pytorch.org/进入pytorch官网,因为我的CUDA是12.4版本,所以我下载的版本是CUDA12.1
复制下面的这个代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121回到 Anaconda prompt,将上面的这串代码复制进去:
输入:y
等待它安装成功即可。
至此,pytorch安装结束。
三、验证pytorch安装是否成功
首先,输入 python ,进入 python 环境
python然后,输入 import torch
import torch最后,输入 torch.cuda.is_available(),返回 ture 说明安装成功。
torch.cuda.is_available()