如题,pytorch工程中,在autoinit.py中遇到
pycuda._driver.Error:cuInit failed:unknown error
解决方案,安装nvidia-modprobe包即可:
sudo apt-get install nvidia-modprobe

如题,pytorch工程中,在autoinit.py中遇到
pycuda._driver.Error:cuInit failed:unknown error
解决方案,安装nvidia-modprobe包即可:
sudo apt-get install nvidia-modprobe