首页服务器Web服务器 docker挂载NVIDIA显卡运行pytorch的方法

docker挂载NVIDIA显卡运行pytorch的方法

admin 2019-10-13 12:32:26 阅读() 评论()

写在前面：　　请参考之前的文章安装好CentOS、NVIDIA相关驱动及软件、docker及加速镜像。　　主机运行环境 $ uname -aLinux CentOS 3.10.0-514.26.2.el7.x86_64 #1 SMP T…

写在前面：

　　请参考之前的文章安装好CentOS、NVIDIA相关驱动及软件、docker及加速镜像。

　　主机运行环境

$ uname -aLinux CentOS 3.10.0-514.26.2.el7.x86_64 #1 SMP Tue Jul 4 15:04:05 UTC 2017 x86_64 x86_64 x86_64 GNU/Linux$ cat /usr/local/cuda/version.txtCUDA Version 8.0.61$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2#define CUDNN_MAJOR   6#define CUDNN_MINOR   0#define CUDNN_PATCHLEVEL 21#define CUDNN_VERSION  (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)#include "driver_types.h"# NVIDIA 1080ti