Windows Python3.6环境下安装PyTorch0.4.0
pip安装
CPU版本
pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
GPU版本
目前gpu版本支持cuda8.0,cuda9.0,cuda9.1,请选择版本进行下载安装(不要同时执行下面三条命令)。
pip install http:/...
原创
2018-06-06 07:40:38 ·
1922 阅读 ·
0 评论