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')
python添加 CUDA 环境变量
于 2024-03-02 20:11:26 首次发布