Traceback (most recent call last):
File "/media/lmy/One Touch/weather4cast/weather4cast-2022-main/train.py", line 199, in <module>
main()
File "/media/lmy/One Touch/weather4cast/weather4cast-2022-main/train.py", line 193, in main
train(params, options.gpus, options.checkpoint)
File "/media/lmy/One Touch/weather4cast/weather4cast-2022-main/train.py", line 148, in train
trainer, checkpoint_callback = get_trainer(gpus, config)
File "/media/lmy/One Touch/weather4cast/weather4cast-2022-main/train.py", line 124, in get_trainer
trainer = pl.Trainer(gpus=gpus, max_epochs=max_epochs,
File "/home/lmy/anaconda3/envs/weather4cast/lib/python3.8/site-packages/pytorch_lightning/utilities/argparse.py", line 345, in insert_env_defaults
return fn(self, **kwargs)
File "/home/lmy/anaconda3/envs/weather4cast/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 433, in __init__
self._accelerator_connector = AcceleratorConnector(
File "/home/lmy/anaconda3/envs/weather4cast/lib/python3.8/site-packages/pytorch_lightning/trainer/connectors/accelerator_connector.py", line 212, in __init__
self._set_parallel_devices_and_init_accelerator()
File "/home/lmy/anaconda3/envs/weather4cast/lib/python3.8/site-packages/pytorch_lightning/trainer/connectors/accelerator_connector.py", line 529, in _set_parallel_devices_and_init_accelerator
raise MisconfigurationException(
pytorch_lightning.utilities.exceptions.MisconfigurationException: CUDAAccelerator can not run on your system since the accelerator is not available. The following accelerator(s) is available and can be passed into `accelerator` argument of `Trainer`: ['cpu'].
目前是在Ubuntu系统下跑程序,出现这个错误,我不是很明白如何去解决,我在终端中检查cuda是已经安装好的并且显示true,cudnn.h也是安装好的,为什么跑程序就显示用不了呢?cuda版本是10.2.
针对无法使用GPU的问题,可以尝试以下解决方案:
确认GPU是否驱动成功,可以使用nvidia-smi命令查看当前GPU的状态,如果没有显示则需要检查驱动是否安装成功。
确认cuda和cudnn是否安装成功,可以使用查看版本的命令,如/usr/local/cuda/version.txt,或者nvcc -V命令。同时也需要设置环境变量,将cuda添加到PATH中。
确认tensorflow-gpu是否安装成功,可以使用conda install tensorflow-gpu命令安装,同时也需要安装对应的cudatoolkit和cudnn。
如果存在多个版本的cuda,可以通过更改环境变量切换CUDA版本。同时如果需要安装cuDNN,可以下载对应版本后进行安装,在安装过程中需要将相关文件复制到cuda的安装目录下,并设置目录和文件的权限。
综上所述,针对无法使用GPU的问题,需要逐一检查以上几个步骤是否都被正确执行,如果有任何一个步骤存在问题都可能导致无法使用GPU。
系统装的有问题吧。