You can install CUDA on CentOS or Red Hat Linux.

Procedure

  1. Navigate to directory on the virtual machine in which to download the NVIDIA CUDA distribution.
    cd <download_directory>
  2. Download the NVIDIA CUDA package for your version of Linux using the wget command.
    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda- repo-ubuntu1804_10.0.130-1_amd64.deb
  3. Install the CUDA package using the rpm -i command.
    sudo rpm -i cuda-repo-rhel7-10.0.130-1.x86_64.rpm 
  4. Run the yum clean all and yum -y install commands as shown to update your environment and install the CUDA software package.
    sudo yum clean all
    sudo yum -y install cuda-10-0 
  5. To confirm your GPU partition size or verify the resources available on your vSphere Bitfusion deployment, run the NVIDIA System Management Interface (nvidia-smi) monitoring application .
    bitfusion run -n 1 nvidia-smi
  6. Navigate to the directory containing the CUDA Matrix Multiplication (matrixMul ) sample files.
    cd /usr/local/cuda/samples/0_Simple/matrixMul 
  7. Run the make and bitfusion run commands against the matrixMul sample file.
    sudo make
    bitfusion run -n 1 ./matrixMul

What to do next

Install and configure NVIDIA cuDNN. See Install NVIDIA cuDNN.