解決linux服務器上安裝深度學習環境,多CUDA版本問題


問題描述

在實驗室服務器上安裝自己的深度學習環境,但是可能服務器上默認的CUDA,與你需要的版本不同,一般地,CUDA安裝好會在以下目錄查看(包含其他用戶安裝的各個版本)

這時需要在你自己的目錄下修改~/.bashrc文件中的PATH,參考了如下修改方式並未成功

我自己修改的PATH,修改后使用nvcc -V 查看版本還是公用目錄默認的9.0

查看$CUDA_HOME,還是9.0版本

分析原因:因為配置公共環境的時候,默認把CUDA9.0的路徑加到了PATH里面了,這種修改方式沒考慮到原本的PATH路徑里面就有一個CUDA9.0的路徑,可以查看PATH的值

可以看到這里面已經有一個9.0版本的路徑,而且在前面,所以在使用nvcc -V命令或者使用CUDA的時候,首先在9.0的路徑里面搜到了,導致后面的9.2就沒起作用

這里是因為我執行了多次的source命令,導致9.2的路徑在PATH中重復存在了

解決步驟

應該把cuda-9.2的路徑添加在PATH的前面,不是加在后面,在.bashrc做如下修改

ORIGINAL_PATH=$PATH
ORIGINAL_CUDA_HOME=$CUDA_HOME
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-9.2/bin:$PATH
export CUDA_HOME=/usr/local/cuda-9.2
export LD_LIBRARY_PATH=/usr/local/cuda-9.2b64:$LD_LIBRARY_PATH

最后,再執行一下source ~/.bashrc 或者重新登錄,才會生效


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM