说明: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
完成!