1、查找自己的cuda版本:
cat /usr/local/cuda/version.txt
 

 我的是10.0
2、安装对应版本的cupy:
pip install cupy-cuda100
 
100是对应写你的版本号
 
3、添加到环境变量:
export CUDA_PATH=你的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=你的cuda安装路径
export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH
 

pip list查询之前是否安装过,如果有需要先卸载:pip uninstall cupy-cudaxxx,再重新安装。相关推荐