创建分区

conda create -n cpu

安装Cpu版本

https://pytorch.org/

conda install pytorch torchvision torchaudio cpuonly -c pytorch

激活环境

conda activate cpu
验证
退出当前分区

conda deactivate

安装GPU版本

创建分区conda create -n gpu

激活环境

conda activate gpu

安装gpu版本

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

验证

python
import torch
torch.cuda.is_available()

启动jupyter notebook

在刚才配置好的分区中

切换 conda activate gpu

安装jupyter

conda install jupyter notebook

启动jupyter

jupyter notebook