因为tensorflow现有版本暂时不适配CUDA10.2,而通常服务器或者带有显卡的PC不会因为一个项目而更换整个CUDA版本。为了解决这个问题,Tensorflow官方开发人员建议,因为CUDA10.2向下兼容CUDA10.1,因此在Tensorflow推出新的版本之前可以通过修改CUDA库软连接(符号链接)的方式适配现有的Tensorflow版本。
具体方法如下:
Actual 10.2 libcudart code is in
/usr/local/cuda-10.2/
我们决定将CUDA库建立软连接到/usr/lib/x86_64-linux-gnu/
to the libcudart.so.10.2
file.
sudo ln -s /usr/local/cuda-10.2/targets/x86_64-linux/lib/libcudart.so.10.2 /usr/lib/x86_64-linux-gnu/libcudart.so.10.1
网友评论