Linux上配置Pytorch环境

一、安装anaconda

创建一个文件夹方便管理

mkdir /opt/data/private/software-packages

安装wget命令

sudo apt-get update
sudo apt-get install wget

下载Anaconda3,得到一个Anaconda3的.sh脚本文件

wget https://repo.anaconda.com/archive/Anaconda3-2019.10-Linux-x86_64.sh

安装anaconda

bash Anaconda3-2019.10-Linux-x86_64.sh

输入想要安装Anaconda3的位置/usr/lkt/myapp/Anaconda3

安装成功

关闭页面，重启使anaconda生效

二、安装Pytorch

检查python版本

python --version

创建和python版本一致的pytorch环境（版本、环境名可更改）

conda create -n pytorch python=3.7

环境的激活、退出、删除（删除时先退出虚拟环境然后移除该虚拟环境）

conda activate 【环境名称】
conda deactivate
conda remove -n pytorch --all

pip使用清华镜像源

进入pytorch的官网，根据系统、python版本、cuda版本选择下载命令。

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

国内其他pip源

清华：https://pypi.tuna.tsinghua.edu.cn/simple
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学：http://pypi.hustunique.com/
山东理工大学：http://pypi.sdutlinux.org/
豆瓣：http://pypi.douban.com/simple/

安装成功进行测试

python
import torch
torch.cuda.is_available()
# 返回True 接着用下列代码进一步测试
torch.zeros(1).cuda()

三、在Pycharm中配置

setting-->project interpreter-->Add python  interpreter -->New environment选择conda Enviroment–>Existing interpreter浏览–>点击conda Enviroment–>interpreter浏览–>选择Anaconda安装路径下的\envs\pytorch\bin目录下的python.exe–>点击ok–>