根据电脑硬件条件,确定Pytorch的版本?

2023-12-19 21:49:08

根据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
>>>

在这里插入图片描述

文章来源:https://blog.csdn.net/xingshanchang/article/details/135066468
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。