conda instal cudatoolkit=11.0 cudnn -c pytorch -c conda-forge -c nvidia
安装mxnet-gpu
pip install mxnet_cu110
安装nccl(多显卡需要)
conda install nccl -c conda-forge
30系列安装mxnet
最新推荐文章于 2024-04-17 10:50:14 发布
本文详细介绍了如何在30系列硬件上安装MXNet的GPU版本。首先,通过conda安装cudatoolkit和cudnn,然后使用pip安装mxnet_cu110以匹配CUDA版本。对于多显卡环境,还提供了安装nccl的步骤。
摘要由CSDN通过智能技术生成