一、问题
最近学习AI,需要在Ubuntu下安装TensorFlow。结果TensorFlow装好后,import TensorFlow时报错:
Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory
二、网上已有的办法尝试
网上目前已有的解决方法是在命令行中安装cuda。例如
https://blog.csdn.net/qq_28193019/article/details/103146116
https://blog.csdn.net/qq_26018075/article/details/111546689
上面两篇文章的思路相同,一个是装10.0版本,一个是装11.0版本。
但是我尝试了上面的方法后,却行不通。可能是镜像配置有问题吧,速度很慢。因此需要使用下面的离线下载方法。
三、使用离线下载办法
打开cuda的官网下载网址:https://developer.nvidia.com/zh-cn/cuda-downloads
依次选择Linux->x86_64->20.04->deb(local)
走到这里,如果执行官方推荐的命令,可能还是会失败。因此有一个小技巧:在windows中利用迅雷下载,下载这两个文件(尤其是第二个2.4G的大文件)。复制链接,在迅雷中点新建下载任务,下载窗口自动弹出来,点立即下载。
网速快的话几分钟就下载好了。然后利用Xshell工具自带的Xftp,把下载好的文件传到ubuntu(这个步骤很容易,后面有机会再写个文章吧)。
四、安装cuda
其实就是cuda官网的安装方法执行一遍就好了(去掉下载的两个命令)。
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo dpkg -i cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
五、测试
python
import tensorflow
如果看到上面的提示,恭喜,问题解决!