Bootstrap

ubuntu20.04+cuda11.1安装mxnet-gpu

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
;