book
book copied to clipboard
book提供的paddlepaddle/book镜像无法使用GPU进行训练
按照https://github.com/PaddlePaddle/book/blob/develop/README.md#running-the-book 部分的设置是用GPU对模型进行1训练:
- sudo nvidia-docker run -it paddlepaddle/book:latest-gpu /bin/bash 进入该容器;
- 将01.fit_a_line模型train.py中 use_cuda = False改为 use_cuda = True
- python train.py报错信息如下:
其他docker image可以正确使用GPU对该模型进行训练。
- 在该容器卸载paddle后,通过pip install paddlepaddle-gpu重新安装paddle,是用GPU对模型进行训练,报错信息如下:
fixed by @JepsonWong
please use docker image of paddlepaddle/book:latest-gpu-cuda9.0-cudnn7