当前位置: 首页 > news >正文

【Python】环境配置,【Pytorch】GPU版本安装

总结:
使用conda新建切换环境,然后使用pip安装卸载包

  • 【python】pip conda_conda list没有pytorch_myaijarvis的博客-CSDN博客

pip换源

https://blog.csdn.net/maotenghua/article/details/104188086

在当前用户目录下创建pip目录,即C:\Users\xxx\pip\ (xxx指代用户名),在pip目录下创建pip.ini文件,文件内容如下
( C:\Users\jcz\pip )

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host = pypi.tuna.tsinghua.edu.cn

conda换源

https://mirrors4.tuna.tsinghua.edu.cn/help/anaconda/

GPU版本

疑问

【参考:Python深度学习:安装Anaconda与PyTorch库(GPU版本)_哔哩哔哩_bilibili】

【参考:为什么电脑装了pytorch没有安装cuda,还是能够使用gpu? - 知乎】

【参考:Pytorch-gpu版安装教程【注意:无需提前安装cuda和cudnn】_pytorch一定要cuda吗_山风wind的博客-CSDN博客】
在这里插入图片描述

在这里插入图片描述
但下载很容易断,所以请看方法二

方法一

  1. 电脑安装显卡驱动
  2. 手动安装cuda
cmd中输入命令 nvidia-smi
CUDA Version显示的是最高支持的版本+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 531.61                 Driver Version: 531.61       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 3060       WDDM | 00000000:0A:00.0  On |                  N/A |
| 30%   27C    P8               13W / 170W|   1173MiB / 12288MiB |      1%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+输入:nvcc -V  
即可看见自己安装的cuda版本nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Feb__8_05:53:42_Coordinated_Universal_Time_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0

然后去这里https://developer.nvidia.com/cuda-toolkit-archive 下载对应的cuda并安装

  1. 用conda新建环境,使用pip安装pytorch-gpu (使用conda安装老是失败)

pytorch官网:https://pytorch.org/get-started/previous-versions/

conda create -n pytorch python=3.7conda activate pytorchpip install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
# torchvision==0.14.1+cu117 torchaudio==0.13.1  这两个选择性安装,这两个都是才几M的文件# 验证
import torch 
print(torch.cuda.is_available()) # True

方法二(推荐)

建议:后续也可以把这三个包单独下载下来,使用pip分别安装
下载地址:https://download.pytorch.org/whl/cu117
安装命令 pip install 文件路径

pip install C:\Users\jcz\Downloads\torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl

搜索torch-1.13.1+cu117 ,cp37就是python3.7

torchvision和torchaudio可以根据需要再安装,一般是用不上的

技巧:下载的时候复制链接到IDM,然后IDM下载过程中速度会波动,可以点击暂停然后点击开始,这样下载速度会很快!
在这里插入图片描述

在这里插入图片描述

方法三

【参考:完全小白部署GPU版的ChatGLM-6B,类chatGPT程序_哔哩哔哩_bilibili】

下载cuda
https://developer.nvidia.com/cuda-11-6-1-download-archive?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

下载cuDNN
https://developer.nvidia.com/rdp/cudnn-archive

双击下载好的cuda****.exe
将cudnn解压,将bin、include、lib复制和替换到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

conda create -n chatglm python=3.10conda activate chatglm安装pytorch
# pytorch官网直接找
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia  ==需要下载很多东西==
http://www.lryc.cn/news/146120.html

相关文章:

  • BEVFusion复现 (Ubuntu RTX3090)
  • Python基础知识学习与回顾
  • SpringBoot笔记——(狂神说)——待续
  • Linux TCP编程流程
  • pyqt5 QuickStart
  • Qt6 for Windows 环境搭建(Visual Studio)
  • 探索未知世界:桌面端3D GIS引领地理信息新时代
  • 微信小程序 趣味学习与益智游戏系统APP
  • 【单片机】UART、I2C、SPI、TTL、RS232、RS422、RS485、CAN、USB、SD卡、1-WIRE、Ethernet等常见通信方式
  • 【Java从0到1学习】13 Java IO流
  • linux并发服务器 —— 多进程并发(四)
  • 华为云Stack的学习(四)
  • Midjourney 完整版教程(从账号注册到设计应用)
  • 保护香港服务器的方法
  • Redis——》Pipeline
  • Git企业开发控制理论和实操-从入门到深入(六)|多人协作开发
  • 通过 ChatGPT 学习 Python
  • 开发卡牌gamefi游戏需要多少钱?
  • linux服务TCP参数配置
  • 部署Spring Boot项目
  • Java 中数据结构LinkedList的用法
  • jmeter递增压测线程组配置
  • hutool工具
  • Redis.conf 配置文件详解
  • linux磁盘空间满了
  • Ansible项目实战管理/了解项目环境/项目管理
  • hadoop 学习:mapreduce 入门案例一:WordCount 统计一个文本中单词的个数
  • vue2项目中el-input单独使用max和maxlength不生效问题
  • 源码角度看待线程池的执行流程
  • 我们的第一个 Qt 窗口程序