这篇博客介绍了如何在拥有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。
PyTorch 2.8
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
Frank（Zhiyang-Dou）apex安装
dinosaurcity cannot import name ‘amp‘ from “apex“ / from apex import amp
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
您可能感兴趣的与本文相关的镜像
PyTorch 2.8
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
3200
3398
779
1395
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
