pytorch cuda 报错

cuda安装之后使用pytorch报如下错误

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.

解决方法，cuda及pytorch安装完成之后，在.bashrc中添加如下

export  PATH=/usr/local/cuda-11.4/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.4/bin
export CUDA_VISIBLE_DEVICES=0,1
           

如果还是有问题，用sudo apt-get install nvidia-modprobe 安装，安装完成后可以使用

检验cuda是否正常的方法

import torch
flag = torch.cuda.is_available()
print(flag)

输出为：True cuda正常