Ubuntu18.04安装apex遇到的一些问题

@Ubuntu18.04安装apex遇到的一些问题

如果使用conda安装的cudatoolkit

在安装apex的时候会报错找不到nvcc,主要的原因是我的pytorch环境是使用anaconda创建的,而anaconda安装的cudatoolkit中是不包含nvcc的。解决的方法就是从nvidia的官网下载对应版本的toolkit进行安装。
可以参考
https://www.it610.com/article/1291625014214336512.htm
需要注意的就是在安装的时候,如果安装过驱动的话要选择不安装驱动。然后cuda的版本和conda安装的toolkit版本要匹配(这个时候conda支持的可直接安装的最高版本为cuda11.1)

上一篇:FAIR开源框架detectron2代码解析


下一篇:pytorch版本,cuda版本,系统cuda版本查询和对应关系