进入python
1. 查看pytorch版本：
import torch print(torch.__version__)
2.查看cuda版本：
print(torch.version.cuda)
3.查看cuda是否可用
print(torch.cuda.is_available())
4.查看可用cuda数量
print(torch.cuda.device_count())
进入python
1. 查看pytorch版本：
import torch print(torch.__version__)
2.查看cuda版本：
print(torch.version.cuda)
3.查看cuda是否可用
print(torch.cuda.is_available())
4.查看可用cuda数量
print(torch.cuda.device_count())
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
