估計你已經被深度學習環境的安裝和配置整得焦頭爛額了,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/