Torch CUDA相关问题

Linux查看Torch相关参数

查看torch版本

进入自己的虚拟环境 （如resnet）.

（resnet）（项目路径）>python
>>> import torch
>>> print(torch.__version__)

查看torch安装后gpu能否正常调用

进入自己的虚拟环境 resnet.

（resnet）（项目路径）>python
>>> import torch
>>> torch.cuda.is_available()   # gpu能否被torch正常调用

查看cuda版本

进入自己的虚拟环境 resnet.

（resnet）（项目路径）>python
>>> import torch
>>> torch.version.cuda          # 查看环境cuda版本