python下载pytorch(通过pip+清华源)

最新推荐文章于 2024-05-25 18:09:06 发布

淸圆风

最新推荐文章于 2024-05-25 18:09:06 发布

阅读量1w

点赞数 9

文章标签： python pytorch pip

本文链接：https://blog.csdn.net/qq_44464101/article/details/125903136

版权

python下载pytorch(通过pip+清华源)

使用conda下载总是报http错误，下载失败，当尝试去掉 -c pytorch下载的只是cpu版本，而不是gpu版本

cmd界面查看自己电脑支持的最高conda

D:\Tool\Anaconda\envs>NVIDIA-smi
Wed Jul 20 23:08:23 2022
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 511.69       Driver Version: 511.69       CUDA Version: 11.6     |
|-------------------------------+----------------------+----------------------+
| 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 ... WDDM  | 00000000:01:00.0 Off |                  N/A |
| N/A   45C    P8    N/A /  N/A |      0MiB /  4096MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A       484    C+G   ...2txyewy\TextInputHost.exe    N/A      |
|    0   N/A  N/A      2020    C+G   ...tracted\WechatBrowser.exe    N/A      |
|    0   N/A  N/A      2792    C+G   ...e6\promecefpluginhost.exe    N/A      |
|    0   N/A  N/A      3716    C+G   ...5n1h2txyewy\SearchApp.exe    N/A      |
|    0   N/A  N/A     12960    C+G   ...onnectUniversalClient.exe    N/A      |
|    0   N/A  N/A     14900    C+G   ...d\runtime\WeChatAppEx.exe    N/A      |
|    0   N/A  N/A     16184    C+G   ...cw5n1h2txyewy\LockApp.exe    N/A      |
|    0   N/A  N/A     16440    C+G                                   N/A      |
|    0   N/A  N/A     20296    C+G   ...llMobileConnectClient.exe    N/A      |
+-----------------------------------------------------------------------------+

.CUDA Version: 11.6 代表你电脑最高支持的conda版本

pip命令下载

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib==2.2.4

pip使用清华源镜像 ；
pytorch的官网pip

在对应conda环境使用

在这里插入图片描述

验证是否安装pytorch-gpu成功

(python3.6) C:\Users\小黄鸭>python
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.10.2+cu113
>>> torch.cuda.is_available()
True
>>>