https://github.com/williamfalcon/pytorch-gpu-install
Instructions for installing pytorch to run on GPU
https://github.com/williamfalcon/pytorch-gpu-install
Last synced: 11 months ago
JSON representation
Instructions for installing pytorch to run on GPU
- Host: GitHub
- URL: https://github.com/williamfalcon/pytorch-gpu-install
- Owner: williamFalcon
- License: mit
- Created: 2017-12-06T19:07:19.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2019-09-06T08:46:52.000Z (almost 7 years ago)
- Last Synced: 2025-03-15T22:43:53.399Z (over 1 year ago)
- Homepage:
- Size: 6.84 KB
- Stars: 36
- Watchers: 2
- Forks: 9
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Pytorch GPU install instructions
At the end of these instructions you will have:
1. A fully working conda environment running pytorch on GPUs
```bash
# create conda env
conda create -n torch python=3.6
# activate the created environment
conda activate torch
# install numpy
pip install numpy
# install torch (cuda 9)
conda install pytorch torchvision cuda90 -c pytorch
# if cuda 9 fails, install this way using cuda 8
conda install pytorch torchvision -c pytorch
conda install cuda80 -c soumith
# test gpu install
python -c 'import torch; print(torch.rand(2,3).cuda())'
```