在跑一些深度學習代碼的時候,如果需要使用mmdetection3d框架,下載的pytorch的cudatoolkit最好需要和本機的cuda版本是一樣的,即輸入nvcc -V命令后顯示的版本一樣。 但是如果是在學校里,一般是服務器管理員裝的cuda驅(qū)動是啥版本,cudatoolkit就是啥版本,且非
在進行深度學習代碼的開發(fā)時,如果需要使用mmdetection3d框架,最好將下載的pytorch的cudatoolkit版本與本機的cuda版本保持一致。可以通過輸入以下命令來確認版本是否一致:
nvcc -V
在學;蚬镜姆⻊掌魃,一般由管理員安裝cuda驅(qū)動,因此cudatoolkit版本也是由管理員確定,非root用戶無法更改。
但是即便是非root用戶,也可以為自己的用戶下的虛擬環(huán)境更改cudatoolkit版本,本文將介紹如何在虛擬環(huán)境中進行cudatoolkit版本的更改操作。
小編推薦閱讀本站所有軟件,都由網(wǎng)友上傳,如有侵犯你的版權,請發(fā)郵件[email protected]
湘ICP備2022002427號-10 湘公網(wǎng)安備:43070202000427號© 2013~2025 haote.com 好特網(wǎng)