pyrorch与paddle安装教程

pyrorch与paddle安装教程 1.pytorch安装

选择cuda11.1

检测pytorch是否安装成功？
在pytorch环境中，在脚本中输入如下代码：

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

运行后，结果显示如下：

说明pytorch安装成功！

2.paddle安装

选择cuda10.2

检测paddle是否安装成功？
在paddle环境中，在脚本中输入如下代码：

import paddle
print(paddle.utils.run_check())

运行后，结果显示如下：

说明paddle安装成功！

参考：利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装—免额外安装CUDA和cudnn（适合小白的保姆级教学）