1. CUDA installation
1. Query the CUDA driver version:
2. Download and install:
3. Check the version of CUDA in the environment
nvcc -V
Two, Pytorch installation
1. pytorch official website:
2. Select the installation method:
3. Localized installation method:
1. Tsinghua mirror source
2. Local installation
3. Post-installation test
(pytorch-gpu) C:\Users\u_name>python
>>>import torch
>>>torch.__version__
'2.0.1'
>>>torch.cuda.is_available()
True