Shulou(Shulou.com)06/01 Report--

This article mainly shows you "how to install and use PyTorch", the content is easy to understand, clear, hope to help you solve your doubts, the following let the editor lead you to study and learn "how to install and use PyTorch" this article.

I. introduction of the environment

Cat / etc/*release*

OpenEuler release 20.03 (LTS)

Uname-a

Linux 4.19.90-2003.4.0.0036.oe1.x86_64 # 1 SMP Mon Mar 23 19:10:41 UTC 2020 x86 "64 GNU/Linux

Nvidia-smi NVIDIA-SMI 440.100 Driver Version: 440.100 CUDA Version: 10.2

GeForce MX250 0MiB / 2002MiB

Python3.8-version

Python 3.8.6

Currently, torch does not support python3.9.

II. Installation

Get the installation command: https://pytorch.org/get-started/locally/#linux-pip

PyTorch Build: Stable (1.7.0)

Your OS: Linux

Package: Pip

Language: Python

CUDA: 10.2

Run this Command: pip install torch torchvision

Specific command: pip3.8 install torch torchvision

Download all versions from: https://download.pytorch.org/whl/torch_stable.html

III. Verification

Python3.8-c "import torch;x = torch.rand (5,3); print (x)"

Tensor ([[0.4857, 0.5264, 0.5910], [0.9196, 0.8787, 0.5331], [0.1681, 0.0487, 0.2963], [0.3761, 0.2133, 0.2272], [0.0082,0.3242,0.3002])

Python3.8-c "import torch;print (torch.cuda.is_available ())"

True

The above is all the contents of the article "how to install and use PyTorch". Thank you for reading! I believe we all have a certain understanding, hope to share the content to help you, if you want to learn more knowledge, welcome to follow the industry information channel!

Welcome to subscribe "Shulou Technology Information " to get latest news, interesting things and hot topics in the IT industry, and controls the hottest and latest Internet news, technology news and IT industry trends.