Imageのダウンロードと起動
まずはDockerの仮想環境(Image)をダウンロードする。
「$ docker pull ubuntu:18.04」
このようなpullコマンドにてローカルにダウンロードすることが可能となる。
イメージの起動にはrun コマンドを使用する。
「$ docker run -it ubuntu:18.04」
PyTorch がインストール済みのイメージは公開されているので以下のコマンドで起動してみる。
「$ docker run -it pytorch/pytorch」
次に「$ python3」でシェルを起動し、
「>>> import torch」
「>>> torch.cuda.is_available()」
でpytorch をインポートできる。
この様になれば問題なくインポートできている。