NVIDIA GTX 3090安装GPU版pytorch,mxnet和autogluon Ubuntu20.04

说明：Ubuntu20.04，NVIDIA GTX 3090的cudatoolkit必须大于11.0。另外，截止2021年11月02日，pytorch和mxnet均支持的cudatoolkit版本也是11.0，对应的pytorch版本只有1.7.1，mxnet版本mxnet_cu110。有的博客装cudatoolkit==11.2的，那种需要编译，感觉太麻烦了，因此没有采用，这里就是anaconda一次性到底

安装

• Ubuntu下装好NVIDIA专用的显卡驱动(自行百度),在终端输入命令"nvidia-smi",显示如下类似的结果表示安装成功（ps:不同版本可能略有差异，大体界面应该是类似的）

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.74       Driver Version: 470.74       CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:83:00.0  On |                  N/A |
| 53%   27C    P8    52W / 390W |    622MiB / 24267MiB |      4%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

• 新建并且激活环境

conda create -n autogluon python=3.8
conda activate autogluon

• 安装pytorch1.7.1

conda install pytorch=1.7.1 torchvision cudatoolkit=11.0 cudnn -c pytorch -c conda-forge -c nvidia

• 安装mxnet-gpu

pip install mxnet_cu110 

• 安装nccl（多显卡需要）

conda install nccl -c conda-forge

• 安装autogluon

pip install autogluon

完成！