天天看點

檢視pytorch paddle 版本,cuda cudnn

檢視torch 環境配置安裝是否成功:
python
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
exit()


           
檢視paddle 環境配置安裝是否成功:
python
import paddle
print(paddle.__version__)
print(paddle.device.get_device())
print(paddle.device.get_cudnn_version())
import paddle.fluid
paddle.fluid.install_check.run_check()
exit()
           

繼續閱讀