多版本CUDA安装使用

Multi-version CUDA install

不要覆盖之前的CUDA安装,这里在Ubuntu18.04的CUDA-10.2原装基础上,再并行安装了CUDA11-11.1 CUDA-11.3
ubuntu下安装多版本cuda及版本切换教程

  1. 按照这个教程,注意不要安装驱动driver,只安装cuda,而且不要讲cuda设置成当前cuda版本,我们随后在zhshrc文件中指定。
    多CUDA版本共存
  2. zhshrc中使用export 加载CUDA版本
    # cuda version can be configured
    export PATH=/usr/local/cuda-11.3/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
    
  3. 查询CUDA_ARCH_BIN
    git clone https://github.com/NVIDIA-AI-IOT/deepstream_tlt_apps.git
    
    cd deepstream_tlt_apps/TRT-OSS/x86
    
    nvcc deviceQuery.cpp -o deviceQuery
    
    ./deviceQuery
    

cudnn 和 Libtorch

下载参考libtorch与tensorRT安装指南
因为都是头文件,所以都下载压缩文件,然后解压在自己的任意路径就好,如果版本多,不要都安装到/usr

CMakeLists.txt设置

set(CMAKE_PREFIX_PATH /home/wzy/3rdparty/libtorch-cxx11-abi-shared-with-deps-1.10.0+cu113/libtorch/share/cmake/Torch)
set(CUDNN_LIBRARY /home/wzy/softwares/cudnn-11.3-linux-x64-v8.2.1.32/cuda/lib64)
set(CUDNN_LIBRARY_PATH /home/wzy/softwares/cudnn-11.3-linux-x64-v8.2.1.32/cuda/lib64)
set(CUDNN_INCLUDE_PATH /home/wzy/softwares/cudnn-11.3-linux-x64-v8.2.1.32/cuda/include)

OpenCV重新编译

下载新OpenCV

由于我们切换了cuda版本,之前的opencv就会出现调用问题,这里重新下载opencv,并在安装时,注意更改安装路径到另一个目录(CMAKE_INSTALL_PREFIX)

编译

cmake -D CMAKE_BUILD_TYPE=RELEASE \
        -D CMAKE_C_COMPILER=/usr/bin/gcc-7 \
        -D CMAKE_INSTALL_PREFIX=/home/wzy/softwares/opencv4.6.0 \
        -D INSTALL_PYTHON_EXAMPLES=ON \
        -D INSTALL_C_EXAMPLES=OFF \
        -D WITH_TBB=ON \
        -D OPENCV_ENABLE_NONFREE=ON \
        -D WITH_CUDA=ON \
        -D BUILD_opencv_cudacodec=OFF \
        -D WITH_CUDNN=ON \
        -D OPENCV_DNN_CUDA=ON \
        -D ENABLE_FAST_MATH=1 \
        -D CUDA_FAST_MATH=1 \
        -D CUDA_ARCH_BIN=8.6 \
        -D WITH_CUBLAS=1 \
        -D WITH_V4L=ON \
        -D WITH_V4L=ON \
        -D WITH_QT=ON \
        -D WITH_OPENGL=ON \
        -D WITH_VTK=ON \
        -D WITH_GSTREAMER=ON \
        -D OPENCV_GENERATE_PKGCONFIG=ON \
        -D OPENCV_PC_FILE_NAME=opencv.pc \
        -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-4.6.0/modules \
        -D BUILD_NEW_PYTHON_SUPPORT=ON \
        -D BUILD_TESTS=OFF \
        -D BUILD_EXAMPLES=ON \
        -D WITH_OPENGL=ON \
        -D PYTHON3_EXECUTABLE=/usr/bin/python3.6 \
        -D BUILD_opencv_python2=OFF \
        -D BUILD_opencv_python3=ON \
        -D BUILD_TIFF=ON .. \
        -D BUILD_LIST=calib3d,features2d,highgui,dnn,imgproc,imgcodecs,\
cudev,cudaoptflow,cudaimgproc,cudalegacy,cudaarithm,cudacodec,cudastereo,\
cudafeatures2d,xfeatures2d,tracking,stereo,\
aruco,videoio,ccalib,video

指定寻找OpenCV

安装好之后,只需要在CMakeLists.txt指定OpenCV路径即可

set(OpenCV_DIR /home/dji/softwares/opencv4.6.0/lib/cmake/opencv4)
find_package(OpenCV)
  • 22
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您可以在Ubuntu上安装多个版本CUDA。以下是一种方法: 1. 首先,确保您的系统上尚未安装任何版本CUDA。如果已经安装CUDA,请先卸载它。 2. 前往NVIDIA官方网站(https://developer.nvidia.com/cuda-toolkit-archive)并下载您想要安装CUDA版本安装程序。选择适用于您的Ubuntu版本和GPU的相应版本。 3. 打开终端,并导航到您下载的CUDA安装程序所在的目录。 4. 运行以下命令来确保您具有执行权限: ``` chmod +x cuda_<version>_linux.run ``` 5. 然后,运行安装程序: ``` sudo ./cuda_<version>_linux.run ``` 在此命令中,将 `<version>` 替换为您下载的CUDA版本的实际版本号。 6. 安装程序将提示您接受许可协议和选择安装选项。请按照屏幕上的指示进行操作。 7. 当安装程序完成后,您需要配置环境变量。打开终端并编辑 `~/.bashrc` 文件: ``` nano ~/.bashrc ``` 8. 在文件末尾添加以下行,将 `<version>` 替换为您安装CUDA版本的实际版本号: ``` export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 9. 保存文件并退出编辑器。 10. 运行以下命令以使更改生效: ``` source ~/.bashrc ``` 11. 现在,您应该已经成功安装了所需版本CUDA。您可以通过运行以下命令来验证安装: ``` nvcc --version ``` 现在您已经安装了多个版本CUDA。您可以按照上述步骤重复执行以安装其他版本
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值