linux服务器安装cupy出错:CuPy is not correctly installed
1、查找自己的cuda版本:cat /usr/local/cuda/version.txt我的是10.02、安装对应版本的cupy:pip install cupy-cuda100100是对应写你的版本号3、添加到环境变量:export CUDA_PATH=你的cuda安装路径export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATHimpor
·
1、查找自己的cuda版本:
cat /usr/local/cuda/version.txt

我的是10.0
2、安装对应版本的cupy:
pip install cupy-cuda100
100是对应写你的版本号
3、添加到环境变量:
export CUDA_PATH=/disk1/linjialing/cuda(你的cuda安装路径)
export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH

import cupy不报错,成功安装。
注意!cupy只能存在一个版本,可以用pip list查询之前是否安装过,如果有需要先卸载:pip uninstall cupy-cudaxxx,再重新安装。
这个办法有个缺点,就是每次使用都需要【重新添加环境变量】。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐

所有评论(0)