Ubuntu14.04上成功安裝PyTorch的詳細教程
阿新 • • 發佈:2019-02-02
本文安裝依賴的環境為Ubuntu14.04、python2.7、cuda8.0、cudnn7.0、opencv2.4.10,這些環境安裝不再贅述,網上教程很多。
1、更新pip(很重要,為第2步做準備)
unzip pip-master.zip
cd pip-master
sudo python setup.py install
2、安裝PyTorch
sudo pip install torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
注:如果不根據第1步更新pip,則用pip安裝pytorch時會報“torch-0.4.1-cp27-cp27mu-linux_x86_64.whl is not a supported wheel on this platform”的錯誤
3、安裝torchvision
sudo pip install torchvision
注:如果安裝過程中提示需要更新Pillow,但出現網路超時更新失敗,有兩種解決辦法:
(1)可以根據連結手動下載Pillow最新原始碼包,通過原始碼安裝Pillow
(2)或者更新pip源(這個網路介紹較多,這兒不再贅述),或者直接用其他源安裝torch(例如用清華源安裝,sudo pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple)