Answers for "ubuntu install torch"

4

install pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Posted by: Guest on November-22-2020
0

install torch 1.7.1

# CUDA 10.2
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch
Posted by: Guest on October-03-2021

Code answers related to "Shell/Bash"

Browse Popular Code Answers by Language