ubuntu18.04 + CUDA10.1 + caffe + pytorch + pytohn3.6 + anaconda3
# 驅動、CUDA、cudnn的安裝以及環境配置:
1. 驅動的版本需要與顯卡的型號以及操作系統對應(有多個版本選擇的,一個不行就更換成另一個版本再嘗試),對應CUDA安裝版本要對應需要與ubuntu版本對應,cudnn安裝版本要對應需要與CUDA版本對應,具體的可以參考這個鏈接。
https://blog.csdn.net/qq_34067821/article/details/90710192
# caffe的安裝總結:
1. 所有的安裝版本要對應
> 比如ubuntu18或ubuntu17與之前的版本安裝caffe的方式就有不同,(17版本之後的ubuntu只需要直接安裝caffe就好了,不用修改那個配置文件,也不用make runtest那一大堆麻煩操作)具體見官網的描述,所以在安裝CAFFE的時候採用的方式也不一樣。
對於Ubuntu(> = 17.04)安裝預編譯的Caffe,包括caffe本身在內的所有產品都包裝在17.04及更高版本中。要安裝預編譯的Caffe包,只需執行此操作即可
sudo apt install caffe-cuda
sudo apt build-dep caffe-cuda
第二條命令需要你修改sources.list
sudo vim /etc/apt/sources.list 然後把裏面的deb-src的註釋掉
然後最難辦caffe環境就裝好了,可以進入python,import caffe 如果沒有報錯,證明,caffe環境配置成功
具體的可以參考這個鏈接。
https://blog.csdn.net/qq_40732093/article/details/92405896
# pytorch的安裝總結:
1. 不要一直卡在一個人的教程上面思考,檢查一下是否環境配置與他完全相同,換一個人的教程再來嘗試,可能更有效果。
最好選用pip的方式來安裝whl的文件,之前參考另一種方式安裝,下載速度很慢而且很容易中斷。注意與anaconda的虛擬環境相對應。
pip install torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl
pip install torchvision
這樣就可以了,輸入以下命令返回true安裝成功。
import torch as t
print(t.cuda.is_available())
具體的可以參考這個鏈接。
https://blog.csdn.net/wuzhiwuweisun/article/details/82753403
最後:
1. 所有都選用最早期的版本是你自己的冒險,但是其實都選擇最新的版本纔是真的冒險,網上教程很多都是老的版本,遇到的問題就不太一樣。
2. 一般遇到問題的時候,不要不斷重來這個操作,檢查一下自己的操作沒有錯誤的時候,很有可能遇到的這個問題就是必經之路,再去搜索解決的辦法就可以了。