Ubuntu16.04系統自帶python2.7和python3.5,系統自帶的python環境不建議更改。
如果想用python3.6可以手動安裝。
1、python3.6安裝
- 安裝命令
sudo apt-get install software-properties-common
sudo add-apt-repository ppa:jonathonf/python-3.6
sudo apt-get update
sudo apt-get install python3.6
- python版本切換
如果想要將python3.6設置爲默認的python版本,命令如下。最後一個參數指定該選項的優先級,默認選擇最高優先級。
update-alternatives --install /usr/bin/python python /usr/bin/python2.7 1
update-alternatives --install /usr/bin/python python /usr/bin/python3.5 2
update-alternatives --install /usr/bin/python python /usr/bin/python3.6 3
列出系統可用的python版本:
update-alternatives --list python
切換選擇系統默認的python版本:
update-alternatives --config python
2、python3.6對應的pip3安裝
pip 是 Python 包管理工具,該工具提供了對Python 包的查找、下載、安裝、卸載的功能。
需要下載與python版本對應的pip工具,來配置該版本python的環境。
- 安裝命令
curl https://bootstrap.pypa.io/get-pip.py | sudo python3.6
- pip版本檢查
pip3 -V
3、pytorch安裝
pytorch官網:https://pytorch.org/
只有在安裝python3.6對應的pip3工具前提下,輸入該命令行,才能在python3.6中配置torch模塊
- 驗證torch模塊是否安裝完成:
python
>>>import torch
>>>print(torch.cuda.is_available())
返回結果是True,則pytorch(gpu版本)安裝成功。