查看显卡信息

1
nvidia-smi

创建虚拟环境

1
2
3
4
5
conda create -n Pytorch python=3.8
conda activate Pytorch
pip list # 查看当前虚拟环境中的包
conda env list # 查看虚拟环境有哪些
where  python # 查看python解释器在哪里

将jupyter与创建的虚拟环境关联

  • 在base环境下安装Pytorch（不推荐）
  • 在创建好的虚拟环境中安装jupyter
1
2
3
conda activate pytorch
pip install nb_conda 
jupyter notebook

查看pytorch是否可用

1
2
import torch
print(torch.cuda.is_available())