一、检查英伟达驱动和CUDA Toolkit是否正确安装
1.任务管理器性能选项卡中能正确显示显卡型号则表示显卡驱动正确安装
2. CUDA Toolkit会跟随pytorch自动安装
二、虚拟环境的准备
Miniconda — Anaconda documentationhttps://docs.anaconda.com/miniconda/
1.安装anaconda或者miniconda
2.conda create -n pytorch python=3.6 创建名称为pytorch,python版本为3.6的虚拟环境
3.activate pytorch 激活名为pytorch的虚拟环境
三、 pytorch的安装
PyTorchhttps://pytorch.org/
nvidia-smi命令查看驱动版本
CPU版本pytorch安装:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
或
pip3 install torch torchvision torchaudio
GPU版本pytorch安装:
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
或
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
下载速度过慢的可以尝试使用镜像或者下载到本地anaconda目录的pkgs文件夹中再安装
四、检验是否安装成功
import torch 在python环境中导入pytorch包,没有报错则表示pytorch安装成功
torch.cuda.is_available () 返回True则表示可以正确使用GPU