-
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 版本
# 可