Install PyTorch Linux: Step-by-Step Guide
Installing PyTorch on a Linux system typically involves following these steps:
- You can install Anaconda (if not already installed) by downloading the corresponding installation package from the official Anaconda website: https://www.anaconda.com/products/distribution.
- To create a new conda environment (optional), you can use the following command :
conda create -n pytorch_env python=3.7
- To activate the conda environment, you can use the following command:
conda activate pytorch_env
- To install PyTorch and TorchVision, you can use the following command:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
If CUDA support is not needed, you can install the CPU version using the following command:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
- To verify the installation, you can use the following Python code to check if PyTorch has been successfully installed:
import torch
print(torch.__version__)
You should be able to successfully install PyTorch on a Linux system by following the steps above. If you encounter any issues, it is recommended to refer to the official PyTorch documentation or seek help on the community forums.