PyTocrch

PyTorch is a Tensors and Dynamic neural networks in Python with strong GPU acceleration.

PyTorch installation

PyTorch is a very fast library, using all possibilities of computer CPU and CUDA on GPU. Therefore, it is very important to install proper configuration of this library. It is possible to install PyTorch with Conda or PIP.

PyTorch installation for Linux

Conda installation


# CUDA version 10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
# CUDA version 11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# CPU only version
conda install pytorch torchvision torchaudio cpuonly -c pytorch

Or if you prefer to use PIP, you can do the same with PIP


# CUDA version 10.2
pip3 install torch torchvision torchaudio
# CUDA version 11.3
pip3 install torch torchvision torchaudio 
--extra-index-url https://download.pytorch.org/whl/cu113
# CPU only version
pip3 install torch torchvision torchaudio 
--extra-index-url https://download.pytorch.org/whl/cpu

PyTorch installation for Windows

For Windows it is a bit more complicated. CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3


# CUDA version 11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# or with pip
pip3 install torch torchvision torchaudio 
--extra-index-url https://download.pytorch.org/whl/cu113

# CPU only 
conda install pytorch torchvision torchaudio cpuonly -c pytorch
# or with pip
pip3 install torch torchvision torchaudio

Access to GPU

GPU is much faster than CPU for tensor calculations, but it is not always available. Therefore, it is better to write program, which can work with GPU and CPU. This can be done by assigning calculations on different devices.


import torch
torch.cuda.is_available() # True is CUDA available, or False
# selection device for calculations
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
# transfer torch object (x) to a new device
x = x.to(device)

Published: 2022-05-04 19:41:45
Updated: 2022-05-04 19:42:39

Last 10 artitles

2023-09-16 03:58:38 - Removing image background with rembg library
2023-09-14 23:39:27 - Python: f-string for printing variables
2023-09-14 22:20:22 - Python: How to unpack the beginning of the list
2023-09-14 02:51:36 - Converting polar field to cartesian coordinates
2023-09-11 17:08:56 - What is the difference between BeautifulSoup or Scrapy
2023-08-16 23:11:23 - Visualising density cloud with planes in Plotly
2023-08-13 01:48:24 - 2D random distribution with density function by Rejection Sampling techniques
2023-08-11 02:57:28 - Generating random distribution for given probability function, 1d case
2023-08-11 02:40:21 - Tools for molecular modelling
2023-08-11 02:10:20 - Atoms and Molecules



PyTocrch » PyTorch - Calculate one step of gradient PyTorch - Calculate few steps of gradient PyTorch - Neural network for simple regression analysis	My Coding > Programming language > Python > PyTocrch PyTocrch PyTorch is a Tensors and Dynamic neural networks in Python with strong GPU acceleration. PyTorch installation PyTorch is a very fast library, using all possibilities of computer CPU and CUDA on GPU. Therefore, it is very important to install proper configuration of this library. It is possible to install PyTorch with Conda or PIP. PyTorch installation for Linux Conda installation `# CUDA version 10.2 conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch # CUDA version 11.3 conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch # CPU only version conda install pytorch torchvision torchaudio cpuonly -c pytorch` Or if you prefer to use PIP, you can do the same with PIP `# CUDA version 10.2 pip3 install torch torchvision torchaudio # CUDA version 11.3 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 # CPU only version pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu` PyTorch installation for Windows For Windows it is a bit more complicated. CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3 `# CUDA version 11.3 conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch # or with pip pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 # CPU only conda install pytorch torchvision torchaudio cpuonly -c pytorch # or with pip pip3 install torch torchvision torchaudio` Access to GPU GPU is much faster than CPU for tensor calculations, but it is not always available. Therefore, it is better to write program, which can work with GPU and CPU. This can be done by assigning calculations on different devices. `import torch torch.cuda.is_available() # True is CUDA available, or False # selection device for calculations device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu') # transfer torch object (x) to a new device x = x.to(device)` Published: 2022-05-04 19:41:45 Updated: 2022-05-04 19:42:39	Last 10 artitles 2023-09-16 03:58:38 - Removing image background with rembg library 2023-09-14 23:39:27 - Python: f-string for printing variables 2023-09-14 22:20:22 - Python: How to unpack the beginning of the list 2023-09-14 02:51:36 - Converting polar field to cartesian coordinates 2023-09-11 17:08:56 - What is the difference between BeautifulSoup or Scrapy 2023-08-16 23:11:23 - Visualising density cloud with planes in Plotly 2023-08-13 01:48:24 - 2D random distribution with density function by Rejection Sampling techniques 2023-08-11 02:57:28 - Generating random distribution for given probability function, 1d case 2023-08-11 02:40:21 - Tools for molecular modelling 2023-08-11 02:10:20 - Atoms and Molecules 9 popular artitles Python Panda (2418) Python NumPy (2061) Python DICTionary HowTo (1131) Python LIST HowTo (1072) Python (1040) Python Generators (943) Programming language (687) R (673) Databases (655)
© 2020 MyCoding.uk -My blog about coding and further learning. This blog was writen with pure Perl and front-end output was performed with TemplateToolkit.