系统相关
首页 > 系统相关> > Ubuntu18.04安装apex遇到的一些问题

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)

标签:apex,toolkit,cudatoolkit,conda,版本,Ubuntu18.04,安装
来源: https://blog.csdn.net/qq_31359085/article/details/114875260