您好,欢迎来到爱go旅游网。
搜索
您的当前位置:首页Apex安装拆坑

Apex安装拆坑

来源:爱go旅游网

Apex安装拆坑
在服务器上安装nvidia apex的拆坑记录
如果是安装pytorch1.1版本的话,如果你系统的cuda版本和虚拟的cuda版本一致的话,这个时候还会报错,是因为官方clone的apex版本过高,需要回退版本

git clone https://github.com/NVIDIA/apex.git
cd apex
git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
python setup.py install --cuda_ext --cpp_ext --deprecated_fused_adam

pytorch1.4好像回退版本不顶用

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- igat.cn 版权所有 赣ICP备2024042791号-1

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务