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,则证明可用。