Bootstrap

【服务器深度学习环境配置】安装cuDNN8.1.1 CUDA11.2 显卡3090

服务器安装cuDNN

参考链接:CUDA官方安装文档
pytorch安装

CUDA 11.2
显卡 3090
ubuntu 18.04

  1. 下载cuDNN
    在这里插入图片描述

  2. 安装cuDNN(deb文件)
    在这里插入图片描述
    我没有导入秘钥,update时有报错(应该是网络问题),也可以成功安装

  3. 测试cuDNN
    在这里插入图片描述
    第3步编译时报错 fatal error: FreeImage.h: No such file or directory
    解决办法:执行sudo apt-get install libfreeimage3 libfreeimage-dev,重新编译后运行./mnistCUDNN,出现Test passed!至此cuDNN安装完成
    在这里插入图片描述
    在这里能看见cudnn的版本
    在这里插入图片描述

验证深度学习环境

参考:知乎 深度学习环境搭建
如何才能验证深度学习的环境(我的是cuda11.2+cudnn8.1.1+tensorflow-gpu 2.7)是OK的呢?

#执行一下Python脚本进行验证
import tensorflow as tf
print(tf.test.is_built_with_cuda()) 
#若输出为true则表示cuda安装成功
print(tf.config.list_physical_devices('GPU'))   #这一栏我改过
#若输出为true则表示能够使用gpu

在这里插入图片描述

;