美文网首页python学习
2021-01-25 windows 10下Pytorch(cp

2021-01-25 windows 10下Pytorch(cp

作者: 可乐W | 来源:发表于2021-01-25 15:56 被阅读0次

由于电脑存在问题,目前无法安装上Pytorch(GPU版),所以先安装Cpu 版本的额,后期会通过解决再安装Gpu 版的。

安装cpu版的Pytorch ,只需要这两行代码即可:

在cmd 下使用pip 安装,如果在安装torchvision 出现问题时候,可以更新下pip 再安装。

pip install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl

pipinstall https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

为了加速安装,可以在后面加上镜像。

-i https://pypi.tuna.tsinghua.edu.cn/simple

例如:pip install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

测试是否安装成功:

>> import torch

>>torch.__version__     #出现版本号则代表安装成功

结果:

在pip list 查看安装:

相关文章

网友评论

    本文标题:2021-01-25 windows 10下Pytorch(cp

    本文链接:https://www.haomeiwen.com/subject/ggqizktx.html