Docker 安装PyTorch + jupyter
一、安装容器
1.方便起见,直接pull一个pytorch镜像,anibali/pytorch,选择合适的cuda版本。
$ docker pull anibali/pytorch:1.5.0-cuda10.2
参考https://github.com/anibali/docker-pytorch(注意看文档,他只在Ubuntu上测试过!)
2.上述文档中给出的usage中是这样的。
$ docker run --rm -it --init \
--gpus=all \
--ipc=h
原创
2020-10-23 22:08:23 ·
1419 阅读 ·
0 评论