pip install pycuda安装出现问题,解决方案

pycuda 安装时出现以下问题:

ERROR: Command errored out with exit status 1:
   command: /home/ed/anaconda3/envs/cam/bin/python -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-c_yustsl/pycuda/setup.py'"'"'; __file__='"'"'/tmp/pip-install-c_yustsl/pycuda/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(__file__);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' bdist_wheel -d /tmp/pip-wheel-wpyh3swb
       cwd: /tmp/pip-install-c_yustsl/pycuda/
  Complete output (104 lines):
  ***************************************************************
  *** WARNING: nvcc not in path.
  *** May need to set CUDA_INC_DIR for installation to succeed.
  ***************************************************************
 

解决方法:

在~/.bashrc文件中增加:

export CUDA_ROOT=/usr/local/cuda

export PATH=$PATH:$CUDA_ROOT/bin

然后 千万别忘记了要执行命令:source ~/.bashrc,或者重新开一个终端。

然后再用pip install pycuda命令,成功!

  • 8
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值