C
C
calculator1212021-11-03 11:54:13
Python
calculator121, 2021-11-03 11:54:13

How to install pytorch for CUDA on windows 10?

I installed cuda version 11.3 from .

C:\Users\Михаил>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Mon_May__3_19:41:42_Pacific_Daylight_Time_2021
Cuda compilation tools, release 11.3, V11.3.109
Build cuda_11.3.r11.3/compiler.29920130_0

Then in the anaconda virtual environment ran the conda command
install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

But pytorch doesn't see cudo What could be the problem? Maybe I installed cuda in the wrong place (since I didn’t bother and chose what the installer suggested)
torch.cuda.is_available() # False

Answer the question

In order to leave comments, you need to log in

Didn't find what you were looking for?

Ask your question

Ask a Question

731 491 924 answers to any question