环境配置(重要)
版本兼容性
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorchconda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.0 -c pytorchLast updated
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorchconda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.0 -c pytorchLast updated