Ubuntu18.04下安裝深度學習框架Pytorch(GPU加速)
Ubuntu18.04安裝TensorFlow和PyTorch全流程
anaconda
- 去 官網 下載相應版本。
- 在下載位置打開終端。
- 運行sh文件
bash AnacondaXXXXX.sh
。 - 一路yes。
- 重新打開終端,默認使用base環境。
環境:
conda info --e # 查看所有環境
conda create -n XXX python=x.x # 創建以XXX爲名的環境
conda activate XXX # 激活環境
conda deactivate # 回到默認環境
conda remove -n XXX --all # 刪除環境
工具包:
conda install XXXXX #安裝
conda remove XXXXX #卸載
conda list #查看當前環境所有工具包
conda upgrade --all #升級所有工具包
pytorch
Ubuntu16.04環境下PyTorch簡易安裝教程
PyTorch1.0安裝教程(附三個測試示例)
- 去 官網 找到相應版本,將command在終端運行:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
- 在終端打開python,輸入
import torch
import torchvision
print(torch.cuda.is_available())
結果爲true,則證明可用。