pytorch torchvision cuda版本对应关系

如图
pytorch torchvision cuda版本对应关系_第1张图片

conda create -n FSCE python=3.6 pip
conda activate FSCE
conda install -y pytorch=1.4.0 torchvision=0.5.0 cudatoolkit=10.0 -c pytorch
pip install -r requirements.txt

你可能感兴趣的:(pytorch,人工智能,python)