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