How to install PyTorch with conda
PyTorch is an open source machine
learning framework,it is an optimized tensor library for deep learning using
GPUs and CPUs. This tutorials covers steps required to install
PyTorch on windows, Linux and Mac
with conda.
Create new virtual environment
Create virtual environment
pytorch_venv with Python (this post
is written with Python 3.7), using anaconda command prompt
conda create --name pytorch_venv python=3.7
Activate virtual environment
conda activate pytorch_venv
PyTorch installation on Windows using conda
Install PyTorch on Windows for
NON-CUDA devices(CPU)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
Install PyTorch on Windows for
CUDA 10.2 devices
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Install PyTorch on Windows for
CUDA 11.3 devices
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
PyTorch installation on Linux using conda
Install PyTorch on Linux for
NON-CUDA devices(CPU)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
Install PyTorch on Linux for
CUDA 10.2 devices
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Install PyTorch on Linux for
CUDA 11.3 devices
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
PyTorch installation on Mac using conda
Install PyTorch on Mac for
NON-CUDA devices(CPU)
conda install pytorch torchvision torchaudio -c pytorch
Verify installation
import torch
print(torch.__version__)
Category: PyTorch
