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
Topic
Name
8+1 =