Ubuntu18.04下编译安装torchvision

最新推荐文章于 2025-06-01 11:07:38 发布

CV/NLP大虾

最新推荐文章于 2025-06-01 11:07:38 发布

阅读量1.4k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch C++

本文链接：https://blog.youkuaiyun.com/m0_37400316/article/details/119448267

pytorch 同时被 2 个专栏收录

26 篇文章

订阅专栏

C++

12 篇文章

订阅专栏

本文详细介绍了如何在CUDA环境下编译并安装TorchVision。首先下载CUDA和CUDNN相应版本，然后创建torchvision文件夹并进行编译。在编译过程中，需注意避免使用GitHub官方教程的部分命令，以免出现错误。接着，解决CUDNN安装的依赖问题，按照特定顺序安装deb包。最后，通过cmake配置编译路径，并执行`make`与`make install`完成安装。文章还提供了相关资源链接和编译时的环境变量设置示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

NMS

在这里插入图片描述

一、下载源码包

NVIDIA 各种cuda下载包
NVIDIA官方DOC
torchvison官网地址

在这里插入图片描述
本次调用是CUDA，所以下载的的是0.8.1版本

二、编译

新建torchvision文件夹，用来存放最终的安装文件（将来调用是用的include和lib）在这里插入图片描述

mkdir build
cd build

进入build文件夹后打开终端，开始进行编译（这里注意下：不要按照GitHub官方的教程来编译，那个教程的有些命令参数暂时不是很全，会报错的！）首先我们输入如下命令（注意把相关的路径修改为自己的路径）：

cmake -DCMAKE_PREFIX_PATH=/XX/libtorch -DCMAKE_INSTALL_PREFIX=/XX/torchvision -DCMAKE_BUILD_TYPE=Release ..

cuda版本

 cmake -DCMAKE_PREFIX_PATH=/XX/libtorch/  -DCMAKE_INSTALL_PREFIX=/XX/torchvsion -DCMAKE_BUILD_TYPE=Release -DWITH_CUDA=ON ..

-DCMAKE_PREFIX_PATH必须设置，因为torchvision编译时会依赖LibTorch里面的相关文件。

cudnn find失败

在这里插入图片描述

在这里插入图片描述
由于电脑是cuda10.1，故下载的是以上版本。

安装sudo dpkg -i XX.deb进行安装（安装1，2顺序安装）。


安装CUDNN时，有三个安装包要安装，必须先安装libcudnn8_8.0.5.39-1+cuda10.1_amd64.deb，然后才能安装另外两个，不然就会出现以下问题：
dpkg: dependency problems prevent configuration of libcudnn8-dev:
libcudnn8-dev depends on libcudnn8 (= 8.0.5.39-1+cuda10.1); however:
Package libcudnn8 is not installed.

dpkg: error processing package libcudnn8-dev (–install):
dependency problems - leaving unconfigured
Errors were encountered while processing:
libcudnn8-dev

解决方法：
先安装libcudnn8_8.0.5.39-1+cuda10.1_amd64.deb

编译torchvison

make

在这里插入图片描述

make install#安装到（上述cmake指定路径）

在这里插入图片描述

使用

cmake使用

set(CMAKE_PREFIX_PATH "/usr/local/lib;/home/wsx/LibTorch/libtorch150cpu/libtorch;/home/wsx/LibTorch/libtorch150cpu/torchvision060/torchvision")

find_package(TorchVision REQUIRED)

target_link_libraries(app PUBLIC ${TORCH_LIBRARIES} ${OpenCV_LIBS} TorchVision::TorchVision)

makefile

					-I/XX/torchvsion/include/ \

然后就可以在源代码中开心的加入 #include “torchvision/csrc/nms.h”

参考文章

https://blog.youkuaiyun.com/Flag_ing/article/details/109708155
https://blog.youkuaiyun.com/qq_41750952/article/details/111573098
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-linux