报错内容
>>> print(torch.cuda.is_available())
/home/lab/anaconda3/envs/lzp_torch/lib/python3.10/site-packages/torch/cuda/__init__.py:124:
UserWarning: CUDA initialization: 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. (Triggered internally at
/opt/conda/conda-bld/pytorch_1681197222248/work/c10/cuda/CUDAFunctions.cpp:108.)
return torch._C._cuda_getDeviceCount() > 0
False
>>>
解决办法
sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm
文章描述了在使用PyTorch时遇到CUDAinitialization未知错误的情况,警告可能由于环境设置不当,如更改CUDA_VISIBLE_DEVICES。解决方案是通过`sudomodprobenvidia_uvm`和`sudomodprobepnp_nvidia`来修复。
1808

被折叠的 条评论
为什么被折叠?



