多版本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.
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值