【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的版本

安装命令

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())# 是否可用gpuflag = torch.cuda.is_available()print(flag)

来源地址：https://blog.csdn.net/weixin_44764931/article/details/127766816