Install torch1.7

May 12, 2023#深度学习 #windows14

AI Translation

This post is translated from Chinese into English through AI.View Original

AI-generated summary

To install torch1.7, first make sure to install cuda and cudnn. Then, go to the official website and find the appropriate version for your system. For example, if you are using cuda11 on a Windows system, use the following command to install torch: "pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html". If there are no errors, the installation is complete. If the installation is slow, consider changing the pip source.

Installing torch 1.7#

1. Install cuda and cudnn#

For details, please refer to my previous article.

2. Find the version that suits you on the official website#

Previous PyTorch Versions | PyTorch

I installed cuda11 on a Windows system using this command. The installation commands may vary for different systems, please refer to the official documentation for specific instructions.

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

If there are no errors, the installation is complete. If the installation is slow, please consider changing the pip source.