pytorch（1）安装

 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