总结:
使用conda新建切换环境,然后使用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
https://mirrors4.tuna.tsinghua.edu.cn/help/anaconda/
【参考:Python深度学习:安装Anaconda与PyTorch库(GPU版本)_哔哩哔哩_bilibili】
【参考:为什么电脑装了pytorch没有安装cuda,还是能够使用gpu? - 知乎】
【参考:Pytorch-gpu版安装教程【注意:无需提前安装cuda和cudnn】_pytorch一定要cuda吗_山风wind的博客-CSDN博客】
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并安装
去pytorch官网找其他包对应的版本:https://pytorch.org/get-started/previous-versions/
把conda 换成pip即可
conda create -n pytorch python=3.7
conda activate pytorch
pip 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.10
conda activate chatglm安装pytorch
# pytorch官网直接找
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia ==需要下载很多东西==