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

问题描述:先安装了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的安装路径可能不一样。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。