问题及起因
问题
前几天跑项目,笔记本上的GPU可以正常跑起来。要跑VAE模型,重新安装了torch,GPU就无法使用了,我重新安装了 cuda,torch.cuda.is_available()的结果依然是False。
起因
配置项目环境不要重新安装torch,因为在windows环境下,重新安装torch,会是CPU版本的。
解决方法
进入torch官网:https://pytorch.org/get-started/locally/ 选择合适的版本进行安装,可以直接复制下面的Command命令到conda虚拟环境,重新安装torch即可。