Pytorch-RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment_rror: option pytoch/pytorch: is not recognized. pl

作者：羊村懒王 | 2024-03-18 14:17:50

踩

rror: option pytoch/pytorch: is not recognized. please run 'nvidia-smi -h

刚装好GPU版本的Pytorch报了这个错，仔细核对了一下Pytorch的版本和我的cuda版本，是一致的。nvidia-smi运行没有问题
错误如下：

RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, 
e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero.
1
2

然后我按照它给的方式执行了一下

export CUDA_VISIBLE_DEVICES='0,1,2,3
1

结果还是报这个错，最后发现原来是更新了drivers没有重启，
最后reboot了一下，解决！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/263652