【ubuntu下，conda创建的虚拟环境在安装好torch后“import torch”时报错“OSError: libcudnn.so.8: cannot open shared object 】

原创

已于 2024-03-21 17:46:07 修改 · 1.7k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #conda #linux

于 2024-03-21 17:40:34 首次发布

文章讲述了在Ubuntu下使用conda创建的虚拟环境中安装torch后导入时报错的问题，原因在于缺少libcudnn.so.8文件。作者提供了详细的解决方案，包括检查cuda目录、下载并解压cudnn、将相关文件复制到虚拟环境，以及激活虚拟环境并验证问题解决。

ubuntu下，conda创建的虚拟环境在安装好torch后“import torch”时报错“OSError: libcudnn.so.8: cannot open shared object file: No such file or directory”

问题描述

ubuntu下，conda创建的虚拟环境在安装好torch后“import torch”时报错“OSError: libcudnn.so.8: cannot open shared object file: No such file or directory”

我安装的torch使用的命令是

conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia

然后进入python环境时,，import torch报错。

解决方案：

首先查看自己的cuda目录下是否有libcudnn.so.8这个文件，我的cuda目录是“/usr/local/cuda-11.7/targets/x86_64-linux/lib/”没有这个文件，所以可能是缺失cudnn相关的文件，可以把cudnn的lib和include中的文件复制到这个cuda对应的目录下，但是这台服务器的cuda环境不止我在用，还有其他人在使用，为了不影响服务器的整体环境，我将我需要的cudnn文件复制到我的虚拟环境中，具体步骤如下：
1）下载cudnn
https://developer.nvidia.com/rdp/cudnn-archive
我的cuda版本是11.7的，选择下面的这个版本，选