怎么安装mxnet gpu

以下是在Ubuntu上安装MXNet GPU的步骤：

1. 首先，确保您的GPU驱动程序已正确安装并可用。您可以使用以下命令检查：

   nvidia-smi

如果您能够看到GPU信息，则说明您的驱动程序已正确安装。

2. 使用以下命令安装CUDA：

   sudo apt-get install nvidia-cuda-toolkit

3. 安装cuDNN库。您需要从NVIDIA官方网站下载cuDNN库并按照说明进行安装。在安装完成后，将库文件复制到CUDA安装目录中：

   sudo cp /path/to/cudnn/lib64/* /usr/local/cuda/lib64/
   sudo cp /path/to/cudnn/include/* /usr/local/cuda/include/

4. 安装Anaconda并创建新环境：

   wget https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.sh
   bash Anaconda3-5.2.0-Linux-x86_64.sh
   conda create -n mxnet python=3.6

5. 激活环境并安装MXNet：

   source activate mxnet
   pip install mxnet-cu90

6. 确认MXNet是否正确安装。在Python交互式环境中输入以下命令：

   import mxnet as mx
   print(mx.__version__)

如果您看到MXNet版本号，则说明MXNet已正确安装。

希望这可以帮助您安装MXNet GPU。