Pytorch 使用不同版本的 cuda,跟使用不同版本的cuda进行编译扩展库,其实TensorFlow也是一样,编译Dcn v2例子。升级GCC

本文介绍了Pytorch使用不同CUDA版本的情况,以及如何在编译扩展库如DcnV2时选择合适的CUDA版本。Pytorch自带的cudatoolkit主要是动态链接库,而完整CUDA Toolkit用于编译和开发。在Linux系统中,通过设置CUDA_HOME或修改/usr/local/cuda软链接来切换CUDA版本。编译扩展时,GCC版本也可能影响编译过程,需要适时升级。
摘要由CSDN通过智能技术生成

使用pytorch时我们可能需要对其进行扩展一些功能,例如近随机卷积的DcnV2。此时我们则需要选择适合的cuda版本进行安装并进行编译扩展。由于我们电脑上可能已经安装过多个cuda版本,此时我们可以根据通过设置环境变量进行随意切换使用的cuda版本。如下安装多个版本的的电脑
在这里插入图片描述

  1. 为什么安装了pytorch如果编译扩展库需要再安装cuda库。
    不管我们通过什么进行安装pytorch等深度框架的时候,我们可以发现pytorch都会带有cudatoolkit,如下图所示:
    在这里插入图片描述
    上述安装的cudatoolkit与我们通过Nvidia官方提供的CUDA Toolkit是不一样的。具体而言Nvidia官方提供的cuda包是包含很多驱动程序,cuda依赖库,例子等,也包括了CUDA程序的编译器CUDA 所支持的功能所对应的各式库文件以及它们的头文件等,具体的组成可参考 CUDA Toolkit Major Components.
    实际上,Nvidia 官方提供安装的 CUDA Toolkit 包含了进行 CUDA 相关程序开发的编译、调试等过程相关的所有组件。但对于 Pytorch 之类的深度学习框架而言,其在大多数需要使用 GPU 的情况中只需要
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值