栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

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

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

cmd界面查看自己电脑支持的最高conda
D:ToolAnacondaenvs>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   ...2txyewyTextInputHost.exe    N/A      |
|    0   N/A  N/A      2020    C+G   ...tractedWechatBrowser.exe    N/A      |
|    0   N/A  N/A      2792    C+G   ...e6promecefpluginhost.exe    N/A      |
|    0   N/A  N/A      3716    C+G   ...5n1h2txyewySearchApp.exe    N/A      |
|    0   N/A  N/A     12960    C+G   ...onnectUniversalClient.exe    N/A      |
|    0   N/A  N/A     14900    C+G   ...druntimeWeChatAppEx.exe    N/A      |
|    0   N/A  N/A     16184    C+G   ...cw5n1h2txyewyLockApp.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
  1. pip使用清华源镜像 ;

  2. 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
>>>
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1021819.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号