美文网首页
问题解决:TensorFlow怎么使用Cuda 10.2

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

作者: 躲在风口的猪 | 来源:发表于2021-01-15 16:42 被阅读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的安装路径可能不一样。

    相关文章

      网友评论

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

          本文链接:https://www.haomeiwen.com/subject/fdewaktx.html