import os
cuda_home = '/usr/local/cuda'
os.environ['CUDA_HOME'] = cuda_home
os.environ['PATH'] += os.pathsep + os.path.join(cuda_home, 'bin')
os.environ['CPATH'] = os.environ.get('CPATH', '') + os.pathsep + os.path.join(cuda_home, 'include')
os.environ['LD_LIBRARY_PATH'] = os.environ.get('LD_LIBRARY_PATH', '') + os.pathsep + os.path.join(cuda_home, 'lib64')
# 添加 CUDA 环境变量
最新推荐文章于 2024-04-26 16:31:32 发布