根据CUDA确定Pytorch的版本
1 显卡型号:NVIDIA GeForce GTX 970
2 显卡算力:5.2
https://en.wikipedia.org/wiki/CUDA
3 确定CUDA Runtime
4 看自己的驱动: CUDA Driver Version —— 12.2
nvidia-smi
5 确定使用的版本
前面3中runtime ,小于12.2就可以。
适用 6.5 ~ 12.2
6 去Pytorch官网确定
https://pytorch.org/
在可用的当中,选择最新的就行。
7 安装
首先配置一个虚拟环境:
conda create -n xingshanchangpytorch121
然后用命令行安装即可
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
8检验
(xingshanchangpytorch121) C:\Users\96975>python
Python 3.11.5 | packaged by Anaconda, Inc. | (main, Sep 11 2023, 13:26:23) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>>