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