赞
踩
Python3.7
cuda10.0
用 pip 安装torch,安装方式如下:
pip install torch==1.2.0 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
在此之前,我试了更新pip,镜像加速等等,都不行,然后看到一篇分享顺利解决了,因此记录下来。
pip install --user torch==1.2.0 -f https://download.pytorch.org/whl/torch_stable.html -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
成功!!
运行torch.cuda.is_available()
返回为True。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。