ubuntu16.04+cuda10+pytorch安装

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

ubuntu16.04+cuda10+pytorch安装_第1张图片

其他版本:

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



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

 

你可能感兴趣的:(conda)