pytorch gpu 环境安装

1.虚拟环境创建

conda create -n venv python=3.8 -y
conda activate venv

其中venv为虚拟环境名，-y表示安装过程中，选择y|n时，默认选择y，省得安装过程中手动去选择

2. 查看本机的cuda版本

nvidia-smi

可以看出本机的cuda版本为10.1

3. 安装对应版本的pytorch-gpu

这里假设环境需要安装python=3.8 pytorch=1.7.1 ，其中cuda=10.1,

• 查看cuda,python,torch的版本对应关系参考,
  
• 安装pytorch–gpu,

首页如果没找到对应的版本，点击Previous version of Pytorch查看历史版本

官方默认安装torch、torchvision以及torchaudio三个包，假如不需要torchaudio这个包，就把torchaudio == 0.9.0指令删掉

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

4 检查pytorch-gpu是否安装成功

打开终端，切换到pytorch安装环境下，检测程序如下：

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

可以看到，已成功安装pytorch-gpu