WSL + ubuntu20.04 + cuda-toolkit-11-4 + nvidia-docker

由于我之前安装的ubuntu18.04在拉取triton镜像的时候，出现源文件无法下载的问题，所以这次直接安装ubuntu20.04版本的，后面安装cuda-toolkit-11-4 和 nvidia-docker非常简单

nvidia-docker WSL 安装官网：CUDA on WSL :: CUDA Toolkit Documentation

我就是参考官网来安装的，

首先安装CUDA

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
sudo apt-get update

再安装cuda-toolkit

apt-get install -y cuda-toolkit-11-4

最后测试一下cuda-samples

cd /usr/local/cuda-11.4/samples/4_Finance/BlackScholes
make BlackScholes
./BlackScholes

出现下面结果说明安装成功了

接下来就是安装nvidia-docker 

sudo curl https://get.docker.com | sh

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update

sudo apt-get install -y nvidia-docker2 

sudo service docker stop

sudo service docker start

一段操作完就ok,最后再拉个镜像测试一下

sudo docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark

出现下面结果说明安装的没问题