几种方案解决RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environme

首先确认.bashrc中已经配置好基本的cuda环境

# CUDA enviroment variable
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export CUDA_VISIBLE_DEVICES=0

一、重启电脑

重启过短时间又会莫名其妙的失效。

二、执行两条命令

sudo rmmod nvidia_uvm

sudo modprobe nvidia_uvm

原理未知，亲测有效。

三、安装nvidia-modprobe

sudo apt install nvidia-modprobe

博主测试不起作用，其他人测试有效。