【填坑】（三）：AssertionError: Torch not compiled with CUDA enabled

说明你的项目目前所有的虚拟环境没有搭建CUDA

要么是你没有安装CUDA

要么是你的虚拟环境不是通过CUDA搭建的

比如你是VSCode创建的.venv

那么

pip insatall torch

就会报这个错误。

如果你没有安装CUDA

或者不清楚怎么通过CUDA创建虚拟环境

并安装Pytorch

请参阅我的专栏

《从零开始构建GPT模型》

---

今天这个问题的解决方案是

在VSCode中选择CUDA搭建并安装Pytorch的虚拟环境。

操作是：按f1->选如图选项

选择如图选项

 打开控制台，激活虚拟环境，后面是你的虚拟环境名称。

conda activate yourevnename

运行测试代码

import torch

print(torch.__version__)
print(torch.cuda.is_available())

控制台，注意控制代码文件的路径，下面是保存在项目根目录下。若是其他文件夹下，则需修改为：例如（不用有中文名称）

python ./yourdir/test.py

python test.py

 

 有这个结果，说明就可以了。