1. 程式人生 > >Ubuntu14.04上成功安裝PyTorch的詳細教程

Ubuntu14.04上成功安裝PyTorch的詳細教程

本文安裝依賴的環境為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)