Hyeonsoo Lee

docker에서 gpu사용

2

안녕하세요 nvidia-docker를 이용해서 --device /dev/nvidia0:/dev/nvidia0 --device /dev/nvidiactl:/dev/nvidiactl --device /dev/nvidia-uvm:/dev/nvidia-uvm gpu device들을 연결시켜주고 실행을 시켰는데, python -c "from tensorflow.python.client import device_lib; local_device_protos = device_lib.list_local_devices(); print([x.name for x in local_device_protos if x.device_type...