安装APEX

编写不易如果觉得不错,麻烦关注一下~ 

成功截图:

安装APEX_第1张图片

1. GITHUB连接:

https://github.com/NVIDIA/apex

git 命令下载,如果觉得慢的话,可以直接压缩包下载。git clone 命令可以压缩包下载,-recusive 最好是git命令下载因为它可能会关联其他git包。但是也可以按照它提示的指令,将其他库的包放置当前包的某个目录下。

2. APEX需要torch 模块。所以需要下载当前cuda 版本匹配的torch 版本,

否则会爆出cuda 不匹配错误信息。

cuda 版本查询:

nvidia-smi

 

安装APEX_第2张图片

我这里必须下载于CUDA11.1匹配的版本

https://pytorch.org/get-started/locally/

安装APEX_第3张图片

3.安装成功后,按照命令编译即可

安装APEX_第4张图片

cuda小知识:https://blog.csdn.net/qq_42406643/article/details/109545766

 

你可能感兴趣的:(pytorch)