Issue
Has anyone encountered this?
I tried updating drivers and reinstalling cuda
Cuda Version: 11.4
GPU: GeForce RTX 3060 Laptop(6gb)
OS: Windows 10 home
torch.version: 1.9.0+cpu
Solution
You are using a PyTorch version compiled for CPU, you should install the appropriate version instead:
Using
conda
:conda install pytorch torchvision cudatoolkit=11.1 -c pytorch -c conda-forge
Using
pip
:python -m pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
Answered By - Ivan
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.