如何在docker容器中使用GPU以及使用pytorch

最近刚接触docker，如有不当，请指出。

1. 首先要解决在docker中可以使用GPU的问题。

官方给出了解决办法：https://github.com/NVIDIA/nvidia-docker

具体操作如下:

step 1:确认本机可以检测到NVIDIA卡，通过下列命令看自己GPU卡的型号。

lspci -vv | grep -i nvidia

step 2:新建nvidia-container-runtime-script.sh文件

touch nvidia-container-runtime-script.sh
vi nvidia-container-runtime-script.sh

并写入内容如下：

sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update

执行下面语句，完成NVIDIA Container Runtime的安装：

sh nvidia-contain