在跑一些深度学习代码的时候,如果需要使用mmdetection3d框架,下载的pytorch的cudatoolkit最好需要和本机的cuda版本是一样的,即输入nvcc -V命令后显示的版本一样。 但是如果是在学校里,一般是服务器管理员装的cuda驱动是啥版本,cudatoolkit就是啥版本,且非
在进行深度学习代码的开发时,如果需要使用mmdetection3d框架,最好将下载的pytorch的cudatoolkit版本与本机的cuda版本保持一致。可以通过输入以下命令来确认版本是否一致:
nvcc -V
在学校或公司的服务器上,一般由管理员安装cuda驱动,因此cudatoolkit版本也是由管理员确定,非root用户无法更改。
但是即便是非root用户,也可以为自己的用户下的虚拟环境更改cudatoolkit版本,本文将介绍如何在虚拟环境中进行cudatoolkit版本的更改操作。
小编推荐阅读