https://pytorch.org/get-started/locally/
创建环境
conda create -n pytorch python=3.9
激活
activate pytorch
安装pytorch
https://pytorch.org/
根据cuda的版本
pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html --user
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
测试 cuda
import torch torch.cuda.is_available()
安装画图
python3 -m pip install -U pip python3 -m pip install -U matplotlib
安装 notebook
pip install d2l
打开
进入代码路径，激活环境，打开
jupyter notebook
