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

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

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

话题: 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版本的更改操作。

K
K
封测 安全无广告
赞(0)  03.01 需网络
推荐

最新好玩手游

更多

手游风云榜

更多

资讯阅读

更多


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