nvcc fatal : Value 'sm_20' is not defined for option 'gpu-architecture'

nvcc fatal : Value 'sm_20' is not defined for option 'gpu-architecture'
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1
 
 
解决方案:
在setup.py文件中
 
nvcc_flags = ['-arch=sm_60', '--ptxas-options=-v', '-c', '--compiler-options', "'-fPIC'", "-w","-std=c++11"]

将sm_20 改为 sm_60
 
 
已邀请:

要回复问题请先登录注册