PyTorch中apex安装方式与常见问题总结

89542767 发布于2022-11-14 10:07 / 1524人阅读

　　小编写这篇文章的主要目的，是来给大家介绍关于PyTorch的一些相关知识，其中包含有apex的安装方式，具体的安装方法，会给大家去做一个详细的步骤介绍。那么，咱安装的时候，会遇到什么问题呢？下面就给大家详细解答下。

　　背景：这个库的安装不是像其他的一样的直接使用pip install XXX的形式，而是使用原始的Git方式

　　1、apex

　　这是NVIDIA开发的基于PyTorch的混合精度训练加速神器，能够增加运算速度，并且减少显存的占用

　　2、安装步骤

　　git clone https://github.com/NVIDIA/apex.git
　　cd apex
　　pip install-v--no-cache-dir--global-option="--cpp_ext"--global-option="--cuda_ext"./

　　3、虽然就简单三行命令，但是仍有很多“坑”存在

　　（1）git clone克隆失败————解决方式：直接通过网页下载

　　如果出现这种情况，那我们就选择“曲线救国”。在GitHub中直接下载该库，然后解压。并放到相应的位置上。

　　（2）安装步骤中的第三行命令报错

　　报错内容：RuntimeError:Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.Pytorch binaries were compiled with Cuda 9.0.176.

　　解决方式：其错误意思就是cuda和pytorch的版本不对应，但是通过搜索也发现可以不带--global--option也能用

　　于是，修改第三行命令为：

　　pip install-v--no-cache-dir./

　　最后的最后，希望大家都能成功安装，冲冲冲！

　　综上所述，这篇文章就给大家介绍到这里了，希望可以给大家带来帮助。

文章版权归作者所有，未经允许请勿转载,若此文章存在违规行为，您可以联系管理员删除。

转载请注明本文地址：https://www.ucloud.cn/yun/128340.html

Python服务器创建虚拟环境跑代码

　　本文主要介绍了Python服务器创建虚拟环境跑代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧　　一、前言　　最近忙着在服务器上跑代码　　学习积累了一些经验技巧　　这里用来记录分享给大家　　二、创建虚拟环境用来跑代码　　下面我会以一个实例为模板，学习完之后，再删掉　　不会占用大家的服务器　　1、连接上服务器　　比如我的连...

89542767 2022-11-13 23:38 评论0 收藏0
MaskRCNN-Benchmark(Pytorch版本)训练自己的数据以及避坑指南

摘要：一安装地址版本首先要阅读官网说明的环境要求，千万不要一股脑直接安装，不然后面程序很有可能会报错一定要按上面的说明一步一步来，千万别省略，不然后面程序很有可能会报错二数据准备我要制作的原始数据格式是训练文件在一个一、安装地址：MaskRCNN-Benchmark(Pytorch版本) 首先要阅读官网说明的环境要求，千万不要一股脑直接安装，不然后面程序很有可能会报错！！！ PyTor...

Acceml 2019-07-31 10:22 评论0 收藏0
Oracle APEX 系列文章2：在阿里云上打造属于你自己的APEX完整开发环境 (准备工作)

摘要：本系列文章使用的是阿里云的云服务器，仅仅是因为在国内用阿里云的人比较多而已。在这里我们采用将单独部署到上运行的方式完成安装，这也是比较推荐的方式。本文是钢哥的Oracle APEX系列文章中的其中一篇，完整 Oracle APEX 系列文章如下： Oracle APEX 系列文章1：Oracle APEX, 让你秒变全栈开发的黑科技 Oracle APEX 系列文章2：在阿里云上打...

MAX_zuo 2019-07-25 14:16 评论0 收藏0