ubuntu下明确tensorflow/pytorch/nvidia驱动/cuda/cudnn/python版本间对应关系,查看系统中nvidia显卡驱动/cuda/cudnn版本

1. 查看nvidia显卡驱动/cuda/cudnn版本:

 1 #nvidia显卡驱动
 2 cat /proc/driver/nvidia/version
 3 
 4 #cuda
 5 cat /usr/local/cuda/version.txt
 6 
 7 #cudnn
 8 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
 9 
10 #检测自己是否成功安装nvidia显卡驱动
11 nvidia-smi  #若出现电脑GPU列表,即安装成功
12 或者
13 nvidia-settings #显示你的显卡信息

2.tensorflow/cuda/cudnn/python版本关系:

配置深度学习环境一定要明确版本对应关系,才能事半功倍。

首先,明确cuda与nvidia显卡驱动版本对应:
在这里插入图片描述
其次,tensorflow/cuda/cudnn/python版本对应:
GPU:
在这里插入图片描述
其中,cudnn与cuda之间详细版本对应请查看官网:NVIDIA,Yes!
CPU:
在这里插入图片描述

3. 关于pytorch与cuda之间版本的对应:

请参看官网:Pytorch, Yes!

 

posted @ 2020-05-23 17:25  墨池有雨  阅读(673)  评论(0编辑  收藏  举报