【DL】PyTorch安装指南

Backto PyTorch Index

# Step 1: Check python version
python --version
# Step 2: Check cuda version
nvcc --version
# Step 3: Download pytorch from
https://pytorch.org/get-started/previous-versions/
# Step 4: install
pip or conda

• pytorch downloads: https://pytorch.org/get-started/previous-versions/
• pip 下载安装

pip install torch==1.0.1 -f https://download.pytorch.org/whl/cpu/stable # CPU-only build
pip install torch==1.0.1 -f https://download.pytorch.org/whl/cu80/stable # CUDA 8.0 build
pip install torch==1.0.1 -f https://download.pytorch.org/whl/cu90/stable # CUDA 9.0 build
pip install torch==1.0.1 -f https://download.pytorch.org/whl/cu92/stable # CUDA 9.2 build
pip install torch==1.0.1 -f https://download.pytorch.org/whl/cu100/stable # CUDA 10.0 build

• conda 下载安装

conda install pytorch=0.4.1 cuda90 -c pytorch