建立企业网站需要什么,温室大棚建设 网站及排名转卖,专业外贸网站开发,虾皮购物网站怎么做TOC
第一步
检查cuda是否安装#xff0c;CUDA环境变量是否正确设置#xff0c;比如linux需要设置在PATH#xff0c;window下环境变量编辑看看#xff0c;是否有CUDA
第二步#xff0c;核查python中torch版本
首先查看你环境里的pytorch是否是cuda版本#xff0c;我这…TOC
第一步
检查cuda是否安装CUDA环境变量是否正确设置比如linux需要设置在PATHwindow下环境变量编辑看看是否有CUDA
第二步核查python中torch版本
首先查看你环境里的pytorch是否是cuda版本我这里是conda安装的使用conda list查看是这样 圈出来带cuda那安装的肯定时cuda版本但是torch.cuda.is_available()还是False这种情况就比较费解重点来了我们去核对下python环境中用的是什么 可以很清楚的看到这里用的是cpu版本 所以环境使用的确实是cpu才导致加载不到gpu版本可以卸载torch重新安装一下
conda uninstall pytorch然后从官网重新下载一下 记住这里有个坑如果你配置了多个镜像源这个镜像源如果没有gpu版本可能还是会下载cpu版本 下载下来以后一定要核对一下
第三步核查显卡驱动和cuda版本是否适配
如果上述没有问题那么检查驱动和cuda的版本是否适配
NVIDIA-smi这两个版本是否适配需要在官网查看 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components
