问题解决:TensorFlow怎么使用Cuda 10.2

2021-01-15  本文已影响0人  躲在风口的猪

问题描述:先安装了CUDA 10.2,然后才发现TensorFlow 2.3只支持CUDA 10.1,下一个版本的TensorFlow会直接支持CUDA 11。如果不想大动干戈重装安装CUDA,可以采取下面的办法建一个链接即可:

ln -s /usr/local/cuda-10.2/targets/x86_64-linux/lib/libcudart.so.10.2 /path/libcudart.so.10.1

然后将path添加到$PATH里就行,每个人的libcudart.so.10.2的安装路径可能不一样。

上一篇 下一篇

猜你喜欢

热点阅读