问题及起因

问题

前几天跑项目，笔记本上的GPU可以正常跑起来。要跑VAE模型，重新安装了torch,GPU就无法使用了，我重新安装了 cuda,torch.cuda.is_available()的结果依然是False。

起因

配置项目环境不要重新安装torch,因为在windows环境下，重新安装torch,会是CPU版本的。

解决方法

进入torch官网：https://pytorch.org/get-started/locally/ 选择合适的版本进行安装，可以直接复制下面的Command命令到conda虚拟环境，重新安装torch即可。