pip安装命令:
pip install mxnet_cu111
需要下载以及安装的包:
https://developer.nvidia.com/nccl/nccl-legacy-downloads
根据cuda版本下载,我下载的是:
nccl-local-repo-ubuntu2004-2.8.4-cuda11.1_1.0-1_amd64.deb
使用以下命令安装:
sudo dpkg -i nccl-local-repo-ubuntu2004-2.8.4-cuda11.1_1.0-1_amd64.deb
安装完,按照提示把pub安装一下
sudo apt-key add /var/nccl-local-repo-ubuntu2004-2.8.4-cuda11.1/7fa2af80.pub
然后执行以下命令:
sudo apt install libnccl2