看看版本和能不能跑。
import torch
print(torch.__version__)
print(torch.cuda.is_available())
悲伤。
2.1.2+cpu
False
在尝试啦python换版本,pytorch,CUDA,CUDNN,各种版本换来换去。我放弃了。
问了下朋友。新建一个虚拟环境。。。。
在anaconda里运行
conda create -n mybert python=3.11.5
之后顺着点一下
下载pytorch:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
回到vscode,shift+ctrl+p 快速输入interpreter打开新建的环境
再次运行第一步。芜湖~
2.3.0+cu118
True