PyTorch是当下非常流行的深度学习框架,其强大的功能以及灵活的API使得PyTorch成为了许多研究者和开发者的首选。PyTorch提供了丰富的功能,包括对GPU加速的支持,对各种深度学习任务的帮助,以及对模型训练和推理的强大支持。
当我们安装PyTorch后,为了了解其版本以及其它相关的环境信息,我们需要进行一些重要的查看操作。以下是查看PyTorch及其相关组件版本的方法:
- 查看PyTorch版本
import torchprint(torch.__version__)
- 查看torchvision版本
import torchvisionprint(torchvision.__version__)
torchvision`是PyTorch的一个扩展库,主要用于图像和视频处理,是ResNet、Mask R-CNN等许多图像任务中常用的库。 - 查看CUDA版本
import torchprint(torch.version.cuda)
CUDA是NVIDIA开发的一种编程模型,PyTorch通过CUDA可以利用NVIDIA的GPU进行高效的计算。如果torch.version.cuda返回的是None,那么说明你的环境没有安装CUDA,或者你的PyTorch没有启用CUDA。 - 查看cuDNN版本
cuDNN是NVIDIA开发的一种深度神经网络库,它为GPU加速的深度学习任务提供了许多基础功能。要查看cuDNN的版本,可以这样:import torchprint(torch.backends.cudnn.version())
如果上述命令返回的是None,那么可能是你的环境没有安装cuDNN,或者你的PyTorch没有启用cuDNN。 - 查看PyTorch可用性
要检查你的PyTorch是否可用,可以检查其是否支持CPU或GPU:import torchprint(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
如果上述命令返回的是None,那么可能是你的环境没有安装正确的驱动,或者PyTorch没有正确的配置GPU。 - 查看CUDA…(这一部分的描述被省略了,此处可以补充更详细的内容)
对于查看CUDA相关的其他信息或状态,可以通过以下代码片段进行查看:
```scss
import torch
print(torch.cuda.get_device_name(0)) # 显示第一个GPU的名字
print(torch.cuda.device_count()) # 显示GPU数量
print(torch.cuda.get_device_capability(0)) # 显示第一个GPU的CUDA计算能力版本, 是一个整数, 比如7.0, 7.5, 8.0等. 参考下表获取具体的能力版本信息. 也可以通过 torch.cuda.get_device_properties(device) 来获取更详细的设备信息, 其中device是一个设备对象, 可以是cpu或者cuda设备, 也可以是None获取所有设备的信息. 返回的信息包括名称, 是否支持半精度浮点数(half), 是否支持单精度浮点数(float16), 是否支持32位整数(int32), 是否支持64位整数(int64), CUDA计算能力版本等. 还可以通过device_properties[key]的方式来获取任何关键字key的信息点例如”name”,”supported_fp_16”,等等(每个key是依据《Occurrences summary properties for cuda (OpenCL) C盈酗 CLE沃iley虾 cleaningerr eye sole issues gone instructions 爱)里面所定义的). 这个函数可以帮助你更好的了解你的GPU设备情况。比如你是否可以运行特定的代码或者特定的操作等。请参考下表获取具体的能力版本信息:对于计算能力版本,如7.0对应的是Pascal架构(比如GTX 1080),7.5对应的是Volta架构(比如Tesla V100),8.0对应的是Turing架构(比如RTX 2080)。而7代表的是Kepler架构(比如GTX 780),7带一个点5代表的是Maxwell架构(比如GTX 950M)。如果对于使用不同厂商的GPU或其它疑问或问题,可以参考官方文档或社区论坛。