Example Shell for torch 1.1.0 install-332514

Search

SCRIPT & CODE EXAMPLE

CODE EXAMPLE FOR SHELL

torch 1.1.0 install

# CUDA 9.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

# CUDA 10.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch

Source by pytorch.org #

PREVIOUS	NEXT

Tagged: #torch #install

ADD COMMENT