python
首先在终端输入python 进入python环境,
import torch
1.查看torch版本
print(torch.__version__)
2.查看cuda版本
print(torch.version.cuda)
3.查看cuda是否可用
如果可用,会返回true
print(torch.cuda.is_available())
4.退出python环境
exit()
python
首先在终端输入python 进入python环境,
import torch
print(torch.__version__)
print(torch.version.cuda)
如果可用,会返回true
print(torch.cuda.is_available())
exit()