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
cat /usr/local/cuda/version.txt
我的是10.0
pip install cupy-cuda100
100是对应写你的版本号
export CUDA_PATH=/disk1/linjialing/cuda(你的cuda安装路径)
export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH
pip list
查询之前是否安装过,如果有需要先卸载:pip uninstall cupy-cudaxxx
,再重新安装。