首页 > 教程 > 如何在虚拟环境中更改mmdetection3d框架的cudatoolkit版本

如何在虚拟环境中更改mmdetection3d框架的cudatoolkit版本

时间:2024-05-19 | 来源: | 阅读:110

话题: T Ato K C 环境 3D

在跑一些深度学习代码的时候,如果需要使用mmdetection3d框架,下载的pytorch的cudatoolkit最好需要和本机的cuda版本是一样的,即输入nvcc -V命令后显示的版本一样。 但是如果是在学校里,一般是服务器管理员装的cuda驱动是啥版本,cudatoolkit就是啥版本,且非

在进行深度学习代码的开发时,如果需要使用mmdetection3d框架,最好将下载的pytorch的cudatoolkit版本与本机的cuda版本保持一致。可以通过输入以下命令来确认版本是否一致: nvcc -V

在学校或公司的服务器上,一般由管理员安装cuda驱动,因此cudatoolkit版本也是由管理员确定,非root用户无法更改。

但是即便是非root用户,也可以为自己的用户下的虚拟环境更改cudatoolkit版本,本文将介绍如何在虚拟环境中进行cudatoolkit版本的更改操作。


湘ICP备2022002427号-10湘公网安备:43070202000427号
© 2013~2019 haote.com 好特网