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

完成!

你可能感兴趣的:(AutoML,pytorch,深度学习,人工智能,mxnet)