>>> import torch
# 查看torch版本
>>> torch.__version__
>>> import torchvision
# 查看torchvision版本
>>> torchvision.__version__
# 查看gpu是否可用
>>> torch.cuda.is_available()
# 查看gpu是个数
>>> torch.cuda.device_count()
# 查看当前gpu
>>> torch.cuda.current_device()
# 查看gpu设备名称
>>> torch.cuda.get_device_name(0)
#v1.13.1
# CUDA 11.7
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
我使用的:pip3 install torch torchvision torchaudio【其中cuda为11.7,cudnn为cudnn-linux-x86_64-8.8.0.121_cuda11-archive】