安装配置docker + cuda + pytorch的注意事项

1. 首先要确定当前设备显卡驱动支持的是哪个cuda版本。确定方法可以参照上一篇关于安装cuda + pytorch的注意事项。
2. 需要安装nvidia-container-toolkit如下所示(fedora版本)：

dnf config-manager --add-repo https://nvidia.github.io/nvidia-docker/centos7/nvidia-docker.repo
dnf install nvidia-container-toolkit

systemctl restart docker

3. 启动docker时需要指定--gpus all, 例如：

docker run -t -i --rm --name=cuda \
-v ${PWD}/src:/src --gpus all --network=host cuda:10.1 /bin/bash

4. 在虚拟机中安装pytorch时也需要安装当前设备支持的cuda版本。这里同样可以参考上一篇关于安装cuda + pytorch的注意事项