- 安装Anaconda
关于Anaconda的概述、安装、管理等可参考
Anaconda使用总结-简书
在Anaconda3-5.3.0-Linux-x86_64.sh文件所在文件夹下执行命令:
sudo ./Anaconda3-5.3.0-Linux-x86_64.sh
直接进入Anaconda安装界面,就可以完成安装。
-
安装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 版权协议,转载请附上原文出处链接和本声明。