查看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()