【pytorch】torch1.7.1安装、查看torch版本、GPU是否可用

在conda 虚拟环境下安装torch==1.7.1+GPU版本

本机环境

• CUDA 11.0
• Python 3.7

安装torch1.7.1

官网搜索确认需要下载的对应本机cuda的torch版本，使用在线下载即可，会直接安装好torch、torhvision、torchaudio。
1、官网搜索对应cuda的版本

2、安装命令

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

查看安装版本

import torch
# 检测torch、cuda、cudnn版本
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

# 是否可用gpu
flag = torch.cuda.is_available()
print(flag)