美文网首页
Pytorch安装最稳超详细——(需VPN)

Pytorch安装最稳超详细——(需VPN)

作者: hdjwuxvks | 来源:发表于2023-07-30 13:42 被阅读0次

## 安装GeForce驱动程序

### 下载

[英伟达官网](https://www.nvidia.cn/geforce/drivers/)

手动搜索驱动程序:

产品类型:GeForce

产品系列:是什么选什么,没有就找最接近的(后面还有产品进一步找你的显卡),Notebooks表示笔记本,一般选这个。

产品、操作系统、语言选自己的

下载类型:选Studio就好,没有就选全部或者Game Ready都可以

最后开始搜索

搜索结果一般可以随便选,选最新的或者比较新的都可以。

### 安装

注意!安装之前关闭火绒!

用管理员身份运行

可以自定义,安装图形驱动程序就好,PhysX系统软件一般不用安,Experience不用(安也没事),可以执行清洁安装(反正也没什么配置不是,又不玩游戏)

等待安装成功

## 安装CUDA Toolkit

打开cmd(管理员最好),输入

```

nvidia-smi

```

出来个框框就说明上一步驱动程序安装成功。

### 下载

在框框的右上角有个“CUDA Version:”,记作【CUDA Version】比如12.2,就是你能安装的CUDA Toolkit的最高版本,可以就安那个版本,也可以稍低一点。不用担心和你的pytorch所需的CUDA不同,可以向下兼容的,你下pytorch的时候下所需版本完全没问题;反而,如果你刚好安装你所需版本的CUDA Toolkit,反而可能会有问题,我就有这样的经历。

在下面这个网址下载,把12-2改成你的【CUDA Version】就好,注意是-不是.

```

https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

```

选择你的配置(Windows,10,exe(local))

点击Download就好,很快的。

### 安装

用管理员身份运行

安装同上,Experience也可以不安

等待安装成功

打开cmd(管理员最好),输入

```

nvcc -V

```

能看到版本之类的信息。

## 安装Pytorch

进入[网址](https://pytorch.org/get-started/locally/)

红框框那,选择Stable,Windows,pip(注意要点pip),python,如果有你所需的CUDA版本就点他,没有就选稍高些的版本,或者点击这个[链接](https://pytorch.org/get-started/previous-versions/)进入之前版本下载(Ctrl F 查找,比如版本11.6就搜116,注意要是pip)。

找到链接复制下来

```

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117(框框那)

或者

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116(之前版本那)

```

进入要安装Pytorch的虚拟环境,在上面的链接后面加你的系统代理,"--proxy http://127.0.0.1:1080" ,特别注意系统代理的端口,(用pip安装其他包也可以这么做)

```

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 --proxy http://127.0.0.1:1080

```

### 验证

```

python

import torch

torch.cuda.is_available()

```

返回True,说明成功

***

修曦所著,转载注明

如果能帮到人就更好啦

相关文章

网友评论

      本文标题:Pytorch安装最稳超详细——(需VPN)

      本文链接:https://www.haomeiwen.com/subject/kxdspdtx.html