这篇博客介绍了如何在拥有3090显卡的环境下，配置CUDA11.1，并成功安装torch 1.9.0+cu111、torchvision 0.10.0+cu111和torchaudio 0.9.0。随后，通过git clone和pip安装的方式详细阐述了NVIDIA apex的安装过程，包括两种安装命令。同时，博客中还提及了一个常见问题：导入amp时出现错误，提示无法从apex导入amp。
成功安装
参考：
Frank（Zhiyang-Dou）apex安装
dinosaurcity cannot import name ‘amp‘ from “apex“ / from apex import amp
硬件：3090
环境配置： cuda11.1
命令：
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
或者
python setup.py install
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
