conda安装的cudatoolkit与Nvidia官方提供的cudatoolkit的区别

阅读: 评论:0

conda安装的cudatoolkit与Nvidia官方提供的cudatoolkit的区别

conda安装的cudatoolkit与Nvidia官方提供的cudatoolkit的区别

conda安装的cudatoolkit 与Nvidia官方提供的cudatoolkit是不一样的。

1、实际上,Nvidia 官方提供安装的 CUDA Toolkit 包含了进行 CUDA 相关程序开发的编译、调试等过程相关的所有组件。

2、但对于 Pytorch 之类的深度学习框架而言,其在大多数需要使用 GPU 的情况中只需要使用 CUDA 的动态链接库支持程序的运行( Pytorch 本身与 CUDA 相关的部分是提前编译好的 ),不需要重新进行编译过程。在安装了 cudatoolkit 后,只要系统上存在与当前的 cudatoolkit 所兼容的 Nvidia 驱动,就可以直接运行。

3、在大多数情况下,上述 cudatoolkit 是可以满足 Pytorch 等框架的使用需求的。但对于一些特殊需求,如需要为 Pytorch 框架添加 CUDA 相关的拓展时( Custom C++ and CUDA Extensions ),需要对编写的 CUDA 相关的程序进行编译等操作,则需安装完整的 Nvidia 官方提供的 CUDA Toolkit.

4、系统上只需要存在cuda的动态链接库,即可编译安装pytorch(只安装了nvidia驱动,没有官方安装cuda),因此nvidia-smi显示的cuda版本,只是pytorch安装时编译安装的版本,而pytorch运行时真正使用的cuda版本为annaconda为其安装的cudatoolkit中cuda的版本(只安装了nvidia驱动,没有官方安装cuda),这块不明白的,请看:

 

 

 

 

从上图可以看到,系统上的cuda版本为11.1,而conda使用的版本为11.0,根据上面的解释这就不矛盾了!!!

5、安装pytorch的官网上,提供的cuda信息,是安装完成时conda安装的cudatoolkit版本,另外执行cuda版本执行顺序,见:

 

 

6、以上部分是个人见解,如有不对的,请大佬批评指正!!!

本文发布于:2024-01-28 21:54:47,感谢您对本站的认可!

本文链接:https://www.4u4v.net/it/170645009210545.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:区别   官方   conda   cudatoolkit   Nvidia
留言与评论(共有 0 条评论)
   
验证码:

Copyright ©2019-2022 Comsenz Inc.Powered by ©

网站地图1 网站地图2 网站地图3 网站地图4 网站地图5 网站地图6 网站地图7 网站地图8 网站地图9 网站地图10 网站地图11 网站地图12 网站地图13 网站地图14 网站地图15 网站地图16 网站地图17 网站地图18 网站地图19 网站地图20 网站地图21 网站地图22/a> 网站地图23