记录Docker 在Tesla K80 和 RTX 2080配置CUDA,Pytorch过程,RuntimeError: cuda runtime error 11 : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:663解决方案
第一次使用docker,需要跑一个深度学习程序,要求pytorch==1.0.1,因此开始在两台服务器不同显卡上配置docker
Tesla K80
先pull镜像,镜像在https://github.com/anibali/docker-pytorch有说明,直接执行以下命令
docker pull anibali/pytorch:cuda-9.0
然后使用该镜像运行容器,先执行以下命令查看已有镜像
docker images
发现镜像已经被拉入本地:
接下来运行一个container来使用这个镜像
nvidia-docker run -it --name myname -v /home/username:/home/username anibali/pytorch:cuda-9.0 bash
这里使用nivdia-docker来运行,这样才能支持GPU运行。 -it 表示与容器进行交互界面, --name 设置容器名字, -v 建立映射路径,否则无法进入本地路径, anibali/pytorch:cuda-9.0 是镜像名称,bash表示用bash命令进行交互
进入后,由于这个镜像配置的是pytorch 1.0.0ÿ