book icon indicating copy to clipboard operation
book copied to clipboard

book提供的paddlepaddle/book镜像无法使用GPU进行训练

Open JiaXiao243 opened this issue 5 years ago • 2 comments

按照https://github.com/PaddlePaddle/book/blob/develop/README.md#running-the-book 部分的设置是用GPU对模型进行1训练:

  1. sudo nvidia-docker run -it paddlepaddle/book:latest-gpu /bin/bash 进入该容器;
  2. 将01.fit_a_line模型train.py中 use_cuda = False改为 use_cuda = True
  3. python train.py报错信息如下: image 其他docker image可以正确使用GPU对该模型进行训练。
  4. 在该容器卸载paddle后,通过pip install paddlepaddle-gpu重新安装paddle,是用GPU对模型进行训练,报错信息如下: image

JiaXiao243 avatar May 13 '19 06:05 JiaXiao243

fixed by @JepsonWong

JiabinYang avatar Oct 10 '19 05:10 JiabinYang

please use docker image of paddlepaddle/book:latest-gpu-cuda9.0-cudnn7

JiabinYang avatar Oct 10 '19 05:10 JiabinYang