一、在nvidia和cuda安装成功的情况下,运行train.py时出现以下报错:
- RuntimeError: Missing libdevice file
- ImportError: libcupti.so.11.7: cannot open shared object file: No such file or directory
- [Errno 2] No such file or directory: ‘:/usr/local/cuda-11.8/bin/nvcc’
- Unrecognized tensor type ID: AutocastCUDA
- ERROR: Command errored out with exit status 1:
- libNVVM cannot be found. Do
conda install cudatoolkit
: libnvvm.so: cannot - ValueError: Unknown CUDA arch or GPU not supported
这些问题都是由于cuda环境配置没有配置完全,下面是配置方法:
1、打开~/.bashrc文件
vim ~/.bashrc
2、在文件末尾输入:
export PATH=/usr/local/cuda-11.8/bin:$PATH
export CUDA_HOME=/usr/local/cuda-11.8
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64
export NUMBAPRO_NVVM=/usr/local/cuda-11.8/nvvm/lib64/libnvvm.so
export NUMBAPRO_LIBDEVICE=/usr/local/cuda-11.8/nvvm/libdevice/
3、让配置生效
source ~/.bashrc