2022年2月22日
NVIDIA apex安装
摘要： 如果是在docker中安装，步骤如下： （1）docker pull pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel。这里必须是-devel后缀！因为编译时需要NVCC，只有-devel中有NVCC。cuda11.2有问题。 （2）按着NVIDIA/apex中 阅读全文
posted @ 2022-02-22 11:37 kuaqi 阅读(1005) 评论(0) 推荐(0) 编辑