一行命令解決 深度學習環境 TensorFlow PyTorch Keras安裝 cuda cudnn 配置

估計你已經被深度學習環境的安裝和配置整得焦頭爛額了,cuda和cudnn的安裝配置真的好麻煩,下面我將告訴你只要用一行代碼就可以解決這個問題。
不過首先,你得下載安裝anaconda環境,而不是使用純Python軟件。
然後創建一個你想要運行深度學習的環境。
一般使用過Python的人,以上兩步都已經完成,新手的話可以自行查閱如何解決,很簡單。

最後。。。。。。

進入終端,進入自己的環境。。。。。。。

TensorFlow:

conda install tensorflow-gpu

PyTorch:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

anaconda自動爲你選擇所有所需下載和更新的包,包括cuda和cudnn,你只需要等待下載安裝完成即可。

大功告成!

提示

pytorch的安裝命令可以自行查找自己所需的版本:
https://pytorch.org/

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章