Skip to content

williamFalcon/pytorch-gpu-install

Folders and files

NameName
Last commit message
Last commit date

Latest commit

5e7bd3c · Nov 7, 2018

History

11 Commits
Dec 6, 2017
Nov 6, 2018

Repository files navigation

Pytorch GPU install instructions

At the end of these instructions you will have:

  1. A fully working conda environment running pytorch on GPUs
# 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())'

About

Instructions for installing pytorch to run on GPU

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published