先确定 cuda 是否安装成功

nvcc -V

安装成功的话直接在命令行里输入

export CUDA_HOME=/usr/local/cuda

 

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐