wsl安装torch-0.4.0 cpu版本

主要参考:https://padeoe.com/use-pytorch-in-windows-subsystem-for-linux/

目前wsl还是不支持cuda,所以安装的cpu版本。使用pip安装
建议直接下载到本地,否则有可能很慢/下载失败
https://download.pytorch.org/whl/cpu/torch_stable.html
找到cp27mu,我cp27m安装报错

但是换成cp27mu后可以了
命令:

#在whl文件路径下
pip install torch-0.4.0-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

结束后测试发现报错

是因为这个环境缺少future这个包

pip install future

再次测试环境

环境配置完成

猜你喜欢

转载自www.cnblogs.com/lonelyisland/p/12932266.html
今日推荐