Tensorflow cannot open libcuda.so.1

后端 未结 4 1311
感动是毒
感动是毒 2021-02-07 07:27

I have a laptop with a GeForce 940 MX. I want to get Tensorflow up and running on the gpu. I installed everything from their tutorial page, now when I import Tensorflow, I get

4条回答
  •  清歌不尽
    2021-02-07 08:10

    In case anyone still encounters this. First make sure to add the --runtime=nvidia parameter in order to run your container.

    docker run --runtime=nvidia -t tensorflow/serving:latest-gpu
    

    where tensorflow/serving:latest-gpu is the name of the docker image.

提交回复
热议问题