【备忘】python查看torch版本、cuda版本
配置环境相关
·
进入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())
4.28编辑
pytorch 中 GPU 索引与 nvidia-smi 中GPU 编号不一致,原因是nvidia-smi 下的 GPU 编号默认使用 PCI_BUS_ID,而 PyTorch 代码默认情况下设备排序是 FASTEST_FIRST。解决方法为,进入bashrc,添加:
CUDA_DEVICE_ORDER=“PCI_BUS_ID“
终端输入 source ~/.bashrc 使配置生效。
更多推荐
已为社区贡献1条内容
所有评论(0)