1。查看GPU及驱动版本号 查看GPU:命令:lspcigrepinvidia 查看驱动版本sudodpkglistgrepnvidia或者ubuntudriversdevices 查看显卡型号lspcigrepinvidia 查看发布版本号catetcissuelsbreleasea 查看内核版本号unamesrunamea 查看GPU可用性condaactivatetensorflowpythonimporttensorflowastfprint(tf。test。isgpuavailable())如果结果是True,表示GPU可用 查看cuda可用性condaactivatetensorflowpythonimporttensorflowastftf。test。isbuiltwithcuda()print(tf。test。isbuiltwithcuda())返回True表示可用2。查看python、tensorflow版本: condaactivatetensorflowpythonimporttensorflowastf查看版本tf。version或者print(tf。version)查看tensorflow安装路径tf。path或者print(tf。path)3。Ubuntu(Linux)系统下查看自己安装的CUDA和CUDNN的版本 CUDA:(1)catusrlocalcudaversion。txtcat(2)nvidiasmi(3)nvccV(4)condalistgrepcuda CUDNN:(1)catusrlocalcudaincludecudnnversion。hgrepCUDNNMAJORA2(2)condalistgrepcuda4。TensorFlow、Python、CUDA、CUDNN对应版本对应关系:链接:https:tensorflow。google。cninstallsourcelinux5。Ubuntu22。04安装NVIDIA驱动 (1)安装驱动前一定要更新软件列表和安装必要软件、依赖sudoaptgetupdate更新软件列表sudoaptgetinstallgsudoaptgetinstallgccsudoaptgetinstallmake (2)查看GPU型号lspcigrepinvidia (3)官网下载对应驱动(英文路径)官方驱动链接:https:www。nvidia。cnDownloadindex。aspx?langcn (4)卸载原有驱动(第一次安装就不需要了)sudoaptgetremovepurgenvidia (5)禁用nouveau(nouveau是通用的驱动程序)sudogeditetcmodprobe。dblacklist。conf或者(blacklistnouveau。conf) 在打开的blacklist。conf末尾添加如下,保存文本关闭blacklistnouveauoptionsnouveaumodeset0 (6)在终端输入如下更新,更新结束后重启电脑sudoupdateinitramfsu (7)重启后在终端输入如下,没有任何输出表示屏蔽成功lsmodgrepnouveau (8)为了安装新的Nvidia驱动程序,我们需要停止当前的显示服务器。sudotelinit3或者按CtrlAltF1F6中的一个(分别对应进入tty1tty6)) 输入用户名(终端下前面就是用户名)和密码 退出文本界面到图形界面sudotelinit5或者CtrlAltF1F7F8 (9)在文本界面中,禁用Xwindow服务,在终端输入sudoetcinit。dgdm3stop (10)进入到存放驱动的目录下,输入命令:sudochmod777NVIDIALinuxx8664525。89。02。run给你下载的驱动赋予可执行权限,才可以安装sudo。NVIDIALinuxx8664525。89。02。run安装 (11)安装结束后输入输入sudoreboot重启。 (12)驱动安装好后,终端输入nvidiasmi检查是否装好,出现类似下方这样,就好了。 然后输入nvidiasettings调出设置界面,类似下方这样,就OK了。 6。下载安装CUDA (1)在官网下载cuda链接:https:developer。nvidia。comcudatoolkitarchive (2)选择runfile的方式安装 按照下载引导,进行下载和安装 详细过程选择:continueaccept在此之前已经装过显卡驱动了,所以需要取消Driver选项。install (3)配置环境变量sudovim。bashrc或者sudovi。bashrc 打开文件后在文末添加环境(注意cuda文件名,根据情况写)exportPATHPATH:usrlocalcuda11。7binexportLDLIBRARYPATHLDLIBRARYPATH:usrlocalcuda11。7lib64exportCUDAHOMECUDAHOME:usrlocalcuda11。7 刷新环境source。bashrc (4)验证是否安装成功nvccV 像这种情况就是成功了 7。安装CUDNN的方法 (1)下载CUDA版本对应的CUDNN版本:https:developer。nvidia。comrdpcudnnarchiveacollapse805111https:developer。nvidia。comrdpcudnndownload (2)注册完成后,找到对应版本进行下载 (3)参照官方文档进行安装:https:docs。nvidia。comdeeplearningcudnninstallguideindex。htmlinstalllinuxtar tarxvfcudnnlinuxx86648。8。0。121cuda11archive。tar。xzsudocpcudnnarchiveincludecudnn。husrlocalcuda11。7includesudocpcudnnarchiveincludecudnn。husrlocalcuda11。7includesudochmodarusrlocalcuda11。7includecudnn。husrlocalcuda11。7lib64libcudnn (4)检查版本及验证catusrlocalcuda11。7includecudnnversion。hgrepCUDNNMAJORA2 这是小编在学习过程中整理的,如有错误请大家多多指正,文章就分享到这里,希望对大家有帮助!