张天浩
|
新手刚接触docker,用来训练深度学习模型 docker run --name digits -d -p 8080:5000 -v /path/to/this/repository:/data/repo kaixhin/digits但在digits中调用的是CPU过程太慢,于是寻找到了nvidia-docker,开始了奇怪的旅程~~~ nvidia-docker很奇怪的问题,安装完后利用 docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi 验证安装结果,显示为 ===============================================================
bc@bc-pc:~$ nvidia-smi
bc@bc-pc:~$ docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
Tue Apr 30 02:21:43 2019 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 418.56 Driver Version: 418.56 CUDA Version: 10.1 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 GeForce GTX 1070 Off | 00000000:01:00.0 On | N/A | | 0% 44C P0 35W / 166W | 148MiB / 8116MiB | 0% Default | +-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+ | Processes: GPU Memory | | GPU PID Type Process name Usage | |=============================================================================| +-----------------------------------------------------------------------------+ ================================================================ 但是用nvidia-docker去训练模型时
nvidia-docker run --name digits -d -p 8080:5000 -v /path/to/this/repository:/data/repo kaixhin/digits结果还是显示的用CPU完成的,不知道是我的nvidia-docker的用法不对么?我直接在原来的语句前加了nvidia-。 还有,我在用systemctl start nvidia-docker启动nvidia-docker的服务时,提示如图2错误,这个路径根本不存在。 =============================================================
bc@bc-pc:~$ sudo systemctl start nvidia-docker
============================================================= ==============================================================
bc@bc-pc:~$ sudo docker version
================================================================
|
2019-04-30 10:41:36