創建分區

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