Setting up NVIDIA Container for using CUDA 10 on CentOS 7
2 min readJan 30, 2020
Prerequisites
sudo yum update -y
sudo yum install pciutils wget epel-release dkms -y# verify gpu device
lspci | grep -i nvidia# gcc compiler is required for development using the cuda toolkit.
gcc --version# install docker
curl sSL http://get.docker.io | shsudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
Install NVIDIA Driver and CUDA 10
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpmsudo rpm -i cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpm
sudo yum clean all# install cuda & drivers
sudo yum -y install nvidia-driver-latest-dkms cuda
sudo yum -y install nvidia-driver-cuda
sudo yum -y install cuda-drivers# setup your paths
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
sudo ldconfig
Reboot your machine
poweroff
enable hardware [all feature, primary GPU]# verify Nvidia CUDA installation
nvidia-smi
nvcc -V
Install NVIDIA Container Toolkit
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo# install nvidia-container-toolkit
sudo yum install -y nvidia-container-toolkit
sudo systemctl restart docker# run docker with gpu options
docker run --gpus all nvidia/cuda:10.0-base nvidia-smi