Open sonfire186 opened 4 years ago
TF GPU it`s work
@sonfiree are you using the pre-built docker image? Make sure you have installed nvidia-docker, and launch the TASO docker image via run_docker.sh tasoml/cuda100
.
I am using nvidia-docker. I use Build TASO Runtime or Docker Images?
You don't need to build the docker image.
Once you have cloned the TASO github repo, you can directly launch our prebuilt docker image via /path/to/taso/docker/run_docker.sh tasoml/cuda100
.
I have Nvidia-Docker running and I need to run TASO. I do not need to create another container
@sonfiree Just to double check that we are on the same page: when you got the above error messages, were you trying to build a TASO docker image through docker build
, or were you trying to run our pre-built docker using the run_docker
script.
I'm already in the container.
It seems cmake cannot automatically find CUDA libraries to build the runtime. Can you change set(USE_CUDA on)
to set(USE_CUDA /usr/local/cuda/)
and also make sure libcublas.so
and libcudnn.so
are in the folder /usr/local/cuda/lib64
? They should be there in our docker image.
docker image nvidia