@TOC
因为跑一个多目标跟踪的代码,要用到yolov3,所以作为pytorch用户的我,第一次用上了tensorflow。
但是我使用的镜像环境,是cuda10.1版本的:
在这里插入图片描述
代码使用的tensorflow,是1.13.1版本的;需要将cuda降到10.0。
这里有一篇博客,展示了Tensorflow不同版本要求与CUDA及CUDNN版本对应关系,一般我们都是装好了tensorflow-gpu,然后需要对应版本的CUDA及CUDNN,可以让大家参考一下。

用conda安装对应的cudatoolkit

1
conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

用conda安装对应的cudnn

1
conda install cudnn=7.5.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

这样每次运行时,会链接到这个版本的cuda和cudnn,实际的cuda版本没有变。

参考链接:
tf-gpu报错:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
Ubuntu下使用conda在虚拟环境中安装CUDA、CUDNN及Tensorflow
conda安装cudnn和cudatoolkit