logo

PyTorch 环境

王哲峰 / 2022-07-11

目录

Windows 安装 PyTorch

CPU version

使用 pip 安装 PyTorch

$ pip3 install torch torchvision torchaudio

使用 conda 安装 PyTorch

$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

使用 Docker 安装 PyTorch

$ TODO

GPU-CUDA version

使用 pip 安装 PyTorch

# CUDA 11.3
$ pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
# CUDA 11.6
$ pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
# CUDA 11.8
$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# CUDA 12.1
$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

使用 conda 安装 PyTorch

# CUDA 11.3
$ conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# CUDA 11.6
$ conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
# CUDA 11.8
$ conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
# CUDA 12.1
$ conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

使用 Docker 安装 PyTorch

$ TODO

macOS 安装 PyTorch

使用 pip 安装 PyTorch

$ pip3 install torch torchvision torchaudio

使用 conda 安装 PyTorch

$ conda install pytorch::pytorch torchvision torchaudio -c pytorch

使用 Docker 安装 PyTorch

$ TODO

Linux 安装 PyTorch

CPU version

使用 pip 安装 PyTorch

$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

使用 conda 安装 PyTorch

$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

使用 Docker 安装 PyTorch

$ TODO

GPU-CUDA version

使用 pip 安装 PyTorch

# CUDA 10.2
$ pip3 install torch torchvision torchaudio
# CUDA 11.3
$ pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
# CUDA 11.6
$ pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
# ROCm 5.1.1
$ pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.1.1

# CUDA 11.8
$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# CUDA 12.1
$ pip3 install torch torchvision torchaudio
# ROCm 5.6
$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.6

使用 conda 安装 PyTorch

# CUDA 10.2
$ conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
# CUDA 11.3
$ conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# CUDA 11.6
$ conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
# CUDA 11.8
$ conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
# CUDA 12.1
$ conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

使用 Docker 安装 PyTorch

$ TODO

Verification

>>> import torch
>>> x = torch.rand(5, 3)
>>> print(x)

tensor([[0.3380, 0.3845, 0.3217],
	[0.8337, 0.9050, 0.2650],
	[0.2979, 0.7141, 0.9069],
	[0.1449, 0.1132, 0.1375],
	[0.4675, 0.3947, 0.1426]])

>>> import torch
>>> torch.cuda.is_available()

官网安装介绍