Pytorch深度学习环境搭建-Anaconda版

  • Post author:
  • Post category:其他


  1. 安装Anaconda

关于Anaconda的概述、安装、管理等可参考

Anaconda使用总结-简书

在Anaconda3-5.3.0-Linux-x86_64.sh文件所在文件夹下执行命令:

sudo ./Anaconda3-5.3.0-Linux-x86_64.sh

直接进入Anaconda安装界面,就可以完成安装。

  1. 安装pytorch

    在安装之前,需要在Anaconda下创建虚拟环境,执行如下命令:
conda create -n your_env_name python=2.7
conda remove -n your_env_name --all  ##删除所创建的虚拟环境
conda info --env或conda env list  ## 查看创建的虚拟环境列表
conda list #查看包
conda install/remove package_name  #安装/删除包

conda remove -n env_name --all  # 删除虚拟环境 env_name为虚拟环境名

注: your_env_name是将要创建的虚拟环境的名字,自定义。这里我的电脑安装的python为2.7版本。

接着执行如下命令激活虚拟环境:

 source activate your_env_name

添加Anaconda的TUAN镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

进入虚拟环境,开始pytorch的安装。在官网选择操作系统和安装环境执行如下命令进行安装。

在这里插入图片描述

使用pip进行安装时,可以加上清华源:

https://pypi.tuna.tsinghua.edu.cn/simple

安装完成以后,在虚拟环境下,输入python,进入python环境,输入如下命令:

import torch
torch.cuda.is_available()

如果

import torch

没有问题,则说明pytorch安装完成。否则,安装失败。如果,

torch.cuda.is_available()

输出

true

,则GPU安装完成。



版权声明:本文为weixin_37970694原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。