Linux 下终于成功安装 pytorch ！（ Virtualenv 和 Anaconda 安装）

• Virtualenv 安装

1. 安装virtualenv

pip install virtualenv

2. virturalenv 创建虚拟环境

virtualenv py_pytorch
# py_pytorch 为虚拟环境的名字
# 默认保存在当前路径

3. 激活虚拟环境

# 切换到创建环境时的路径！
souce py_pytorch/bin/activate

# 查看当前路径下的文件，可以看到当前路径下有没有自己创建的虚拟环境
ls

# 如果是创建在根目录下，要注意区分：
1. [xxx@localhost ~]$
2. [xxx@localhost /]$

4. torch 安装

# 进入官网，选择与自己 cuda 版本相匹配的版本
# https://pytorch.org/
# CUDA 10.1
pip install torch==1.4.0 torchvision==0.5.0

5. 检查是否安装成功

import torch
# 不报错及成功
# 若安装的 torch 为 GPU 版，还应当输入
torch.cuda.is_available()
# 若输出为 True 及成功

6. 退出虚拟环境

deactivate

注：用 pip 命令安装各种需要的包

• Conda 安装

1. 安装 Anaconda ，具体步骤 优快云 可查

2. 创建虚拟环境

conda create --name py_torch python=3.6
# py_torch 为虚拟环境名称，可指定 python 版本

# 可