GPU版 CPU版 Pytorch安装教程_安装gpu版本的pytorch

前言

安装Pytorch前请先安装Anaconda, 用Anaconda先安装Python虚拟环境，并且根据Pytorch官网最新提示

Python版本最好是3.8-3.11之间，后期有变化都能在Pytorch官网首页看到

如果项目有明确要求请安装对应版本Python，其次建议安装3.8版本Python，低版本适用性更强一些 ，安装虚拟conda环境命令如下

conda create -n pytorch python=3.8 # 创建名称为pytorch并且python为3.8版本的虚拟环境
conda activate pytorch # 进入pytorch虚拟环境
1
2

​​​​​​​​​​​​​​​​​​​​​

GPU版Pytorch安装

GPU版Pytorch版本用的比较多，安装前先确定自己电脑的显卡版本，有两种方法：

1. 第一种方法是nvidia-smi命令可以查看电脑GPU驱动信息

nvidia-smi
1

图中右上角标红框的地方显示的是驱动CUDA版本，并不是算力CUDA的版本，如果这个数字小于11.7，请更新显卡驱动，升级到11.7或以上，因为官网推荐版本为11.7.

如果低于11.7.可以参考Pytorch官网早期版本安装方法。

2. 另一种查看GPU信息的方法是nvcc -V命令

nvcc -V​
1