### 安装 CUDA 10.2 以支持 Python 3.6 环境 为了确保 CUDA 10.2 能够顺利安装并兼容 Python 3.6 环境，建议按照以下方法操作： #### 更新包管理器和安装必要工具 更新系统的软件包列表，并安装一些必要的开发工具和支持库。这一步骤对于后续的 CUDA 安装至关重要。 ```bash sudo apt-get update && sudo apt-get upgrade -y sudo apt-get install build-essential cmake git unzip pkg-config libopencv-dev ``` #### 添加 NVIDIA 的官方仓库密钥以及设置源地址 通过添加 NVIDIA 提供的安全签名密钥来验证下载文件的真实性，并配置 APT 使用 NVIDIA GPU 驱动程序存储库。 ```bash wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.2.89-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1804_10.2.89-1_amd64.deb sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub sudo apt-get update ``` #### 安装 CUDA Toolkit 10.2 及其依赖项 执行此命令可以完成 CUDA 工具链及相关组件的安装过程。注意这里选择了 `--no-install-recommends` 参数减少不必要的额外软件被一同安装进来。 ```bash sudo apt-get install --no-install-recommends cuda-10-2 ``` #### 设置环境变量以便于访问新安装好的 CUDA 库路径 编辑用户的 shell profile 文件（如 `.bashrc`,`.zshrc`），追加如下两行内容到文件末尾处，使当前终端会话能够立即生效这些更改；如果希望永久化修改，则需重启计算机或重新加载该配置文件。 ```bash export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} source ~/.bashrc # 或者 source ~/.zshrc 如果使用的是 zsh 终端 ``` #### 测试 CUDA 是否成功安装 编写一段简单的 C++ 程序用于测试已安装版本能否正常工作。创建名为 test_cuda.cu 的文件并将下面这段代码粘贴进去保存后编译运行它即可查看结果。 ```cpp #include <stdio.h> int main() { printf("Hello, CUDA!\n"); int deviceCount; cudaGetDeviceCount(&deviceCount); printf("Number of devices: %d\n", deviceCount); for (int i = 0; i < deviceCount; ++i) { cudaDeviceProp prop; cudaGetDeviceProperties(&prop, i); printf("Device Number: %d\n", i); printf(" Device name: %s\n", prop.name); printf(" Memory Clock Rate (KHz): %d\n", prop.memoryClockRate); printf(" Memory Bus Width (bits): %d\n", prop.memoryBusWidth); printf(" Peak Memory Bandwidth (GB/s): %.1f\n\n", 2.0 * prop.memoryClockRate * (prop.memoryBusWidth / 8) / 1.0e6); } return 0; } ``` 编译上述代码片段: ```bash nvcc -o test_cuda test_cuda.cu ./test_cuda ``` 以上步骤完成后应该能够在基于 Python 3.6 的环境中正常使用 CUDA 10.2 版本了[^1]。