天天看点

ubuntu18.04 + anaconda + pycharm + pytorch(GPU)

目录

    • anaconda
    • pytorch
    • pycharm

Ubuntu18.04下安装深度学习框架Pytorch(GPU加速)

Ubuntu18.04安装TensorFlow和PyTorch全流程

anaconda

Ubuntu18.04 安装 Anaconda3

  1. 去 官网 下载相应版本。
  2. 在下载位置打开终端。
  3. 运行sh文件

    bash AnacondaXXXXX.sh

  4. 一路yes。
  5. 重新打开终端,默认使用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安装教程(附三个测试示例)

  1. 去 官网 找到相应版本,将command在终端运行:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
           
  1. 在终端打开python,输入
import torch
import torchvision
print(torch.cuda.is_available())
           

结果为true,则证明可用。

pycharm

Ubuntu 18.04 安装 PyCharm