* Follow #223 * Fix #222
@ -69,7 +69,7 @@ If you want to launch container with GPU support, you need to set some environme
.. code-block:: bash
export CUDA_SO="$(\ls /usr/lib64/libcuda* | xargs -I{} echo '-v {}:{}') $(\ls /usr/lib64/libnvidia* | xargs -I{} echo '-v {}:{}"
export CUDA_SO="$(\ls /usr/lib64/libcuda* | xargs -I{} echo '-v {}:{}') $(\ls /usr/lib64/libnvidia* | xargs -I{} echo '-v {}:{}')"
export DEVICES=$(\ls /dev/nvidia* | xargs -I{} echo '--device {}:{}')
docker run ${CUDA_SO} ${DEVICES} -it paddledev/paddle:gpu-latest
@ -68,7 +68,7 @@ def define_py_data_source(file_list, cls, module,
file_list_name = 'train.list'
if isinstance(cls, TestData):
file_list_name = 'test.list'
with open(file_list_name, 'r') as f:
with open(file_list_name, 'w') as f:
f.writelines(file_list)
file_list = file_list_name