首先，配置好Anaconda环境，可参考Anaconda + VScode | Python环境安装与管理。在此基础上，本文介绍Pytorch深度学习环境配置。

Pytorch安装

打开Anaconda Prompt, 查看电脑支持的CUDA版本：

1
nvidia-smi

nvidia-smi

可以看到，我的显卡型号为NVIDIA GTX1660 SUPER，支持的CUDA版本最高到12.2

记得切到你想下载pytorch的环境：

1
conda activate seismic

到Pytorch官网选择低于你电脑支持的cuda版本，复制下载命令，例如我的电脑支持到12.2，这里我选择了11.8：

1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Pytorch

然后漫长的等待,等到出现successful installed，就安装好啦。

检查是否安装成功

打开一个jupter，选择刚刚安装好pytorch的那个环境，import torch 检查是否安装成功。

1
2
3
4
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)

The End

最后，如果你的电脑木有显卡，可以试试在线租卡，比如AutoDL。