创建分区
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