(默认cuda版本和pytorch,以及nvidia-driver匹配)1
. 首先进入python环境,通过torch.cuda.is_available()或者torch.cuda.device_count()来确认cuda是否安装,也可以在命令行里面使用nvcc -V来查看cuda是否安装
2. 如果上述操作没问题,检查代码里面是否有这个:
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
把自己电脑上的cudaid放到里面去
(默认cuda版本和pytorch,以及nvidia-driver匹配)1
. 首先进入python环境,通过torch.cuda.is_available()或者torch.cuda.device_count()来确认cuda是否安装,也可以在命令行里面使用nvcc -V来查看cuda是否安装
2. 如果上述操作没问题,检查代码里面是否有这个:
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
把自己电脑上的cudaid放到里面去