pytorch linux GPU安装

选在对应的环境 https://pytorch.org/

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html -i http://mirrors.aliyun.com/pypi/simple/

pip install torch==1.3.1+cu100 torchvision==0.4.2+cu100 -f https://download.pytorch.org/whl/torch_stable.html -i http://mirrors.aliyun.com/pypi/simple/

posted @ 2020-06-18 11:08  pergrand  阅读(1861)  评论(0)  编辑  收藏  举报