はじめに

機械学習に使用していたサーバが死にました

Ubuntu 18.04.4 Desktop (64-bit)
NVIDIA Geforce GTX 1080
NVIDIA Driver 440.82
CUDA 10.2(2020/06/12現在 PytorchがCUDA11に未対応な為)

SSH Serverのインストール(optional)

$ sudo apt install -y openssh-server
$ sudo systemctl enable ssh
$ sudo systemctl restart ssh

$ docker pull nvcr.io/nvidia/pytorch:20.03-py3

$ docker run --gpus all -it --rm -v local_dir:container_dir --name pytorch nvcr.io/nvidia/pytorch:xx.xx-py3
root@XXXX:/workspace $ python hoge.py

それにしてもDocker凄いですね
面倒くさかった環境構築が一瞬でできるようになるなんて
バックアップはしっかり取りましょう