docker19.03で最速でGPU機械学習環境を作る - Qiita

はじめに nvidiaのGPUを使った機械学習の環境を整えたい。でも、なんか、面倒臭そう... いえいえDockerを使えば簡単です。 下記のコマンドを使うと簡単に環境構築ができます。次に詳しく見ていきましょう。 docker run -v $PWD:/working -e DISPLAY=$DISPLAY --net host --gpus all --shm-size=8gb --name ml -it --workdir /working gcr.io/kaggle-gpu-images/python:v80 /bin/bash