但是A卡并不支持CUDA,虽然可以将CUDA代码转为OpenCL,但是想要适配A卡的生产环境极其麻烦,这边不者高考虑A卡。
AMD的方法是,直接通过代码迁移,让AMD独显也能支持CUDA生态。在最新的版本中,开发者可以轻松的将CUDA文件转换称HIP文并袭件,对应的API也能一对一转换。
(1)先查看斗兆氏当前服务器下的显卡情况: nvidia-smi
(2)用如下命令指定空散猜笑某张卡,来运行程序:
CUDA_VISIBLE_DEVICES=4 python main.py --cuda
没有显卡只有驱动不能运如茄绝行cuda。根据查询相关公开资料显示运行CUDA应用程纳让序需要系统至少有一个支持CUDA的GPU和一个与CUDAToolkit兼容的显卡驱动,安装显卡驱动才能对cuda进行命令的渣姿运行。欢迎分享,转载请注明来源:内存溢出
评论列表(0条)