最近在搞tensorflow的一些东西,话说这东西是真的皮,搞不懂。但是环境还是磕磕碰碰的搭起来了
其实本来是没想到用码头工人的,但是就一台配置较好的服务器,还要运行公司的其他环境,vmware esxi用起来太费劲,还是算了。
<强>环境:
系统:CentOS7 7.4 1708
显卡:英伟达1080 ti
<强>下载所有需要的东东
1, docker-ce百胜回购:https://download.docker.com/linux/centos/docker-ce.repo
2, nvidia-docker百胜回购:https://nvidia.github.io/nvidia-docker/centos7/x86_64/nvidia-docker.repo
3, nvidia cuda百胜回购:http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda -回购rhel7 9.1.85 - 1. - x86_64.rpm
4, nvidia cudnn: https://developer.nvidia.com/cudnn
这个东西需要注册nvidia账号,就不给直接下载地址了。
5, nvidia驱动:http://www.nvidia.cn/Download/index.aspx& # 63; lang=cn
按自己的显卡型号下载
6, nvidia码头工人文件:https://hub.docker.com/r/nvidia/cuda/
这里面可以看到很多dockerfile,选择
9.0 -base-centos7(9.0/基地/Dockerfile)
其他的cuda9.1这些应该也可以用,另外有像重击和运行时这样的,其实就是百胜安装的cuda包不太一样,没多大关系。
点进去后复制下来保存为Dockerfile文件,但是之后搞的时候发现有点问题,修改了一下,可以从这儿复制
从centos: 7
标签维护者“NVIDIA公司& lt; cudatools@nvidia.com>”
运行NVIDIA_GPGKEY_SUM=d1be581509378368edeec8c1eb2958702feedf3bc3d17011adbf24efacce4ab5,,\
curl https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/7fa2af80.pub -fsSL | sed '/^版/d '比;/etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA,,\
echo " $ NVIDIA_GPGKEY_SUM/etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA”| sha256sum - c——严格
# cuda副本。回购/etc/yum.repos.d/cuda.repo
ENV CUDA_VERSION 9.0.176
ENV CUDA_PKG_VERSION——这里9 CUDA_VERSION-1美元
#运行yum安装- y \
# cuda-cudart - CUDA_PKG_VERSION美元,,\
# ln - s cuda - 9.0/usr/local/cuda,,\
# rm射频/var/cache/yum/*
# nvidia-docker 1.0
标签com.nvidia.volumes.needed=" nvidia_driver "
标签com.nvidia.cuda.version=" $ {CUDA_VERSION}”
运行回声“/usr/地方/nvidia/lib”在祝辞/etc/ld.so.conf.d/nvidia。相依,,\
回声“/usr/地方/nvidia/lib64”在祝辞/etc/ld.so.conf.d/nvidia.conf
ENV/usr/local/nvidia/bin路径:/usr/地方/cuda/bin: $ {PATH}
ENV LD_LIBRARY_PATH/usr/local/nvidia/lib:/usr/地方/nvidia/lib64
# nvidia-container-runtime
ENV NVIDIA_VISIBLE_DEVICES所有
ENV NVIDIA_DRIVER_CAPABILITIES计算,效用
ENV NVIDIA_REQUIRE_CUDA“cuda>=9.0”
<强>所有的文件
(root@localhost nvidia) # pwd/root/nvidia
(root@localhost nvidia) #噢
总计420000
drwxr-xr-x。2根根4096 2月10是centos-gpu
-rw-r——r。1根根3335年1月29日北京时间cuda -回购rhel7 9.1.85 x86_64.rpm——1.
-rw-r——r。1根根348817823 2月6 16:26 cudnn - 9.0 - linux - x64 v7.tgz
-rw-r——r。1根根2424年2月9日北京时间docker-ce.repo
-rw-r——r。796年2月9日17:11 nvidia-docker.repo 1根根
-rwxr-xr-x。1根根81242220 1月31日十四19 NVIDIA-Linux-x86_64-390.25.run
centos-gpu里有Dockerfile文件
<强>准备工作
直接上命令,一看就明白
(root@localhost nvidia) # cp docker-ce。回购nvidia-docker。回购/etc/yum.repos.d/(root@localhost nvidia) # rpm -ivh cuda -回购rhel7 9.1.85 x86_64.rpm——1.
(root@localhost nvidia) # yum安装epel-release
(root@localhost nvidia) # yum安装gcc gcc-c + +
(root@localhost nvidia) # yum安装内核*
之前
<>强安装驱动
(root@localhost nvidia) #回声“黑名单”产业的祝辞的在/etc/modprobe.d/blacklist.conf
(root@localhost nvidia) # mv/boot/initramfs -美元(uname - r)。img/boot/initramfs - $ (uname - r)。img。贝克dracut - v/boot/initramfs -美元(uname - r)。img $ (uname - r)
(root@localhost nvidia) 3 # init
(root@localhost nvidia) # chmod + x NVIDIA-Linux-x86_64-390.25.run
(root@localhost nvidia) #。/NVIDIA-Linux-x86_64-390.25.run
