【原】如何安装pytorch

表白：黑白圣堂血天使，天剑鬼刀阿修罗。 
讲解对象：/如何安装pytorch
作者：融水公子 rsgz
===

首先要查看自己的GPU cuda版本是多少

nvidia-smi

C:\Users\Administrator>nvidia-smi
Sat Jul  8 15:24:44 2023
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 531.79                 Driver Version: 531.79       CUDA Version: 12.1     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                      TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 4070 Ti    WDDM | 00000000:01:00.0  On |                  N/A |
|  0%   44C    P8               12W / 285W|    468MiB / 12282MiB |     33%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A      2992    C+G   ...CBS_cw5n1h2txyewy\TextInputHost.exe    N/A      |
|    0   N/A  N/A      7084    C+G   ...5n1h2txyewy\ShellExperienceHost.exe    N/A      |
|    0   N/A  N/A      7688    C+G   C:\Windows\explorer.exe                   N/A      |
|    0   N/A  N/A      7964    C+G   ..._8wekyb3d8bbwe\Microsoft.Photos.exe    N/A      |
|    0   N/A  N/A      8508    C+G   ...2txyewy\StartMenuExperienceHost.exe    N/A      |
|    0   N/A  N/A      9092    C+G   ....Search_cw5n1h2txyewy\SearchApp.exe    N/A      |
|    0   N/A  N/A     10760    C+G   ...ekyb3d8bbwe\PhoneExperienceHost.exe    N/A      |
|    0   N/A  N/A     11496    C+G   ...crosoft\Edge\Application\msedge.exe    N/A      |
|    0   N/A  N/A     12064    C+G   ...51-tg\Telegram Desktop\Telegram.exe    N/A      |
|    0   N/A  N/A     15724    C+G   ...oogle\Chrome\Application\chrome.exe    N/A      |
+---------------------------------------------------------------------------------------+
C:\Users\Administrator>

可以看到我的版本是

CUDA Version: 12.1

2 然后访问这个网站，找对应的版本，复制这个命令pip方式进行下载

https:/\pytorch.org/

=== 

公众号：不浪仙人

谢谢大家的支持！可以点击我的头像，进入我的空间浏览更多文章呢。建议大家360doc[www.360doc.com]注册一个账号登录，里面真的有很多优秀的文章，欢迎大家的到来。
---

';