Geeks_Z の Blog Geeks_Z の Blog
首页
  • 学习笔记

    • 《HTML》
    • 《CSS》
    • 《JavaWeb》
    • 《Vue》
  • 后端文章

    • Linux
    • Maven
    • 汇编语言
    • 软件工程
    • 计算机网络概述
    • Conda
    • Pip
    • Shell
    • SSH
    • Mac快捷键
    • Zotero
  • 学习笔记

    • 《数据结构与算法》
    • 《算法设计与分析》
    • 《Spring》
    • 《SpringMVC》
    • 《SpringBoot》
    • 《SpringCloud》
    • 《Nginx》
  • 深度学习文章
  • 学习笔记

    • 《PyTorch》
    • 《ReinforementLearning》
    • 《MetaLearning》
  • 学习笔记

    • 《高等数学》
    • 《线性代数》
    • 《概率论与数理统计》
  • 增量学习
  • 哈希学习
GitHub (opens new window)

Geeks_Z

AI小学生
首页
  • 学习笔记

    • 《HTML》
    • 《CSS》
    • 《JavaWeb》
    • 《Vue》
  • 后端文章

    • Linux
    • Maven
    • 汇编语言
    • 软件工程
    • 计算机网络概述
    • Conda
    • Pip
    • Shell
    • SSH
    • Mac快捷键
    • Zotero
  • 学习笔记

    • 《数据结构与算法》
    • 《算法设计与分析》
    • 《Spring》
    • 《SpringMVC》
    • 《SpringBoot》
    • 《SpringCloud》
    • 《Nginx》
  • 深度学习文章
  • 学习笔记

    • 《PyTorch》
    • 《ReinforementLearning》
    • 《MetaLearning》
  • 学习笔记

    • 《高等数学》
    • 《线性代数》
    • 《概率论与数理统计》
  • 增量学习
  • 哈希学习
GitHub (opens new window)
  • mac

  • win

  • linux

  • ai

    • PyTorch安装
    • conda
    • cuda
      • cuda安装(多版本切换)
      • pytorch寻找可用CUDA的过程
  • IDE

  • 实用工具

  • NAS

  • GitHub

  • Tex

  • Others
  • ai
Geeks_Z
2024-11-17
目录

cuda

cuda安装(多版本切换)

除了软连接,还可以通过配置环境变量(CUDA_PATH )来实现版本切换。 当本机上安装有多个版本cuda时可以通过一下步骤进行管理/版本切换,比如我们要切换使用cuda11.8: sudo vim ~/.bashrc加入以下的内容: export LD_LIBRARY_PATH=LDLIBRARYPATH:/usr/local/cuda−11.8/lib64exportPATH=PATH:/usr/local/cuda-11.8/bin export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.8 修改完毕,保存,并执行"source ~/.bashrc"

使用 LD_LIBRARY_PATH 来控制适用不同版本的 CUDA runtime。 anaconda 也是用 LD_LIBRARY_PATH 来控制CUDA版本的。 在使用 CMake 等编译和运行的过程中,可以根据 CUDA_HOME 或 CUDA_PATH 这个环境变量选择CUDA 版本,而不需要删除、创建软连接

原文链接: https://zhuanlan.zhihu.com/p/91334380 https://blog.csdn.net/qq_42406643/article/details/109545766

pytorch寻找可用CUDA的过程

查找可用的cuda路径 1、环境变量CUDA_HOME 或 CUDA_PATH 2、/usr/local/cuda。(这是软连接)

注:h3c服务器中软连接已删除,如有要自己在用户家目录(.bashrc 中)配置CUDA环境变量(路径:usr/local/cuda-11.8或usr/local/cuda-12.3) 3、which nvcc的上级上级目录 (which nvcc 会在环境变量PATH中找) 4、如果上述都不存在,则torch.utils.cpp_extension.CUDA_HOME为None,会使用conda安装的cudatoolkit,其路径为cudart 库文件目录的上级目录(此时可能是通过 conda 安装的 cudatoolkit,一般直接用 conda install cudatoolkit,就是在这里搜索到 cuda 库的)。 原文链接: https://blog.csdn.net/qq_41094058/article/details/116207333 https://www.cnblogs.com/yhjoker/p/10972795.html

显卡驱动与cuda版本关系 每个cuda都有一个最低要求的显卡驱动版本。 每个显卡驱动有一个最高支持的cuda版本。 nvidia-smi命令显示的cuda版本是该驱动最高支持的cuda版本。(安装显卡驱动后就可以执行nvidia-smi命令)

查看使用的cuda版本: torch.version.cuda 是位于 torch/version.py 中的一个变量, Pytorch 在基于源码进行编译时,通过 https://github.com/pytorch/pytorch/blob/master/tools/setup_helpers/cuda.py 来确定编译 Pytorch 所使用的 cuda 的安装目录和版本号。并不一定是 Pytorch 在实际系统上运行时使用的 cuda 版本,而是编译该 Pytorch release 版本时使用的 cuda 版本。 torch.utils.cpp_extension.CUDA_HOME #输出 Pytorch 运行时使用的 cuda

#AI软件安装
上次更新: 2025/07/06, 13:25:25
conda
IDEA

← conda IDEA→

最近更新
01
优化器
07-04
02
weight_decay
07-03
03
帮助信息查看
06-08
更多文章>
Theme by Vdoing | Copyright © 2022-2025 Geeks_Z | MIT License
京公网安备 11010802040735号 | 京ICP备2022029989号-1
  • 跟随系统
  • 浅色模式
  • 深色模式
  • 阅读模式