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(适合小白的保姆级教学)

猜你喜欢

转载自blog.csdn.net/weixin_44359479/article/details/120872603