常用Linux命令

​

python
#输入库
import torch
#查看版本

print(torch.__version__)


#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()
#退出python
quit()

​

import torch
print(torch.__version__)#查看torch版本

print(torch.version.cuda)#查看cuda版本

print(torch.cuda.is_available())#查看cuda是否可用
#####返回True说明配置成功
print(torch.cuda.device_count())#查看可用cuda数量