torch1.2GPU版本安装

gpu:

pip3.6 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-linux_x86_64.whl

cpu:

pip3.5 install torch==1.2.0+cpu torchvision==0.4.0+cpu -f

猜你喜欢

转载自blog.csdn.net/weixin_43868576/article/details/114258849
今日推荐