将编译 GPU 驱动生成的三个压缩包复制到 setup_gpu.sh
同目录
压缩包 | 说明 |
---|---|
libraries-[DRIVER_VERSION].tar.bz2 | GPU 动态库 |
tools-[DRIVER_VERSION].tar.bz2 | GPU 工具 |
modules-[COREOS_VERSION]-[DRIVER_VERSION].tar.bz2 | GPU 驱动 |
sudo bash ./setup_gpu.sh COREOS_VERSION DRIVER_VERSION
以 DRIVER_VERSION=367.57
和 COREOS_VERSION=1122.2.0
为例:
执行如下命令完成 GPU 配置:
sudo bash ./setup_gpu.sh 1122.2.0 367.57
查看 GPU 信息
# nvidia-smi
# docker run -it --rm --privileged cuda /bin/bash
# root@9a1ab2d177b0:/opt/nvidia# ./nvidia-smi