Pytorch环境搭建

一、新建虚拟环境

conda create -n env_name python=3.8

二、激活虚拟环境

conda activate env_name

三、安装torch、torchvision

pytorch官网查看版本匹配情况

# CUDA 10.2
conda install pytorch==1.10.0 torchvision==0.11.1 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch

四、查看是否安装成功

import torch
torch.__version__
torch.cuda.is_available()

五、常用命令

1. 退出虚拟环境

conda deactivate

2. 查看现有虚拟环境

conda info -e

3. 查看cuda版本

nvcc -V

4. 查看GPU使用情况

nvidia-smi