跳至主要內容
环境配置

1. 查看 GPU 支持的最高 CUDA 版本

NVDIA 控制面板 -> 左下角系统信息 -> 组件 -> NVCUDA64.DLL产品名称
选择torch版本时选择比该版本号低的版本

2. 下载 cuda

下载链接 cuda 官网

CUDA是一个并行计算平台和编程模型,能够使得使用GPU进行通用计算变得简单和优雅。

nvcc其实就是CUDA的编译器,可以从CUDA Toolkit/bin目录中获取,类似于gcc就是c语言的编译器。

验证是否安装成功 分别运行


HanTx大约 2 分钟深度学习cudaCNNpythontorch