Pytorch_GPU安装

可恶的CSDN审核，竟然不给我的第一篇象征着开始的文章通过审核，那我就写在这里，气死你。
-------------------------------------------------------------------新开始，新气象！-----------------------------------------------------------------------
之前一直抱怨CSDN是垃圾堆，大粪池，想着自己可不能发这种水文章东抄西抄，得发高质量的好文章。谁曾想到现在一篇没发过。。。自己也越来越废物。故从今起发博客，不论简繁，记录自己的探索历程以便将来有需回顾，也有自我督促，自我勉励之意，亦想为之后和我方向一致、迷茫探寻的人一点参考。此贴就是我的新开始，新气象！

Pytorch_GPU版安装

卸载原有CPU版

pytorch安装及卸载_conda 卸载pytorch-CSDN博客

参考了这篇文章，调用了两条命令。

conda uninstall pytorch
conda uninstall libtorch

下面是过程演示

执行第一条命令

C:\Users\hidden-dragon>conda uninstall pytorch
Collecting package metadata (repodata.json): done
Solving environment: done


==> WARNING: A newer version of conda exists. <==
  current version: 23.3.1
  latest version: 23.11.0

Please update conda by running

    $ conda update -n base -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main conda

Or to minimize the number of packages updated during conda update use

     conda install conda=23.11.0



## Package Plan ##

  environment location: C:\Application\anaconda3

  removed specs:
    - pytorch


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    cpuonly-1.0                |                0           2 KB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
    ------------------------------------------------------------
                                           Total:           2 KB

The following packages will be REMOVED:

  libjpeg-turbo-2.0.0-h196d8e1_0
  libuv-1.44.2-h2bbff1b_0
  pytorch-2.1.0-py3.9_cpu_0
  pytorch-mutex-1.0-cpu
  torchaudio-2.1.0-py39_cpu
  torchvision-0.16.0-py39_cpu

The following packages will be SUPERSEDED by a higher-priority channel:

  cpuonly                            pytorch::cpuonly-2.0-0 --> anaconda/cloud/pytorch::cpuonly-1.0-0


Proceed ([y]/n)? y


Downloading and Extracting Packages

Preparing transaction: done
Verifying transaction: done
Executing transaction: done

执行第二条命令

C:\Users\hidden-dragon>conda uninstall libtorch
Collecting package metadata (repodata.json): done
Solving environment: failed

PackagesNotFoundError: The following packages are missing from the target environment:
  - libtorch

也许是因为第一条指令已经把libtorch删除了，所以第二条指令报错找不到libtorch这个待删除的包。

安装GPU版

Pytorch(GPU版)安装 - 知乎 (zhihu.com)

参考了这篇文章

查看GPU版本

CMD中使用如下命令

nvidia-smi

结果如下，重点在于CUDA Version: 12.3该部分

C:\Users\hidden-dragon>nvidia-smi
Mon Dec 25 20:50:56 2023
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 546.01                 Driver Version: 546.01       CUDA Version: 12.3     |
|-----------------------------------------+----------------------+----------------------+
| 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 4060 ...  WDDM  | 00000000:01:00.0  On |                  N/A |
| N/A   38C    P5               6W / 115W |   1438MiB /  8188MiB |     60%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+

执行官网指令

安装需要保证你的N卡的驱动版本号高于torch的最低要求
打开https://pytorch.org/get-started/locally/#windows-prerequisites，选择合适的对应部分，复制得到的命令行语句。我的CUDA是12.3，所以随便选了CUDA12.1，选用的是conda命令。

这里根据参考文章，将-c 后的字符替换为清华源地址即https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
得到命令

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

执行命令，经过漫长的等待，我这里遇到了Solving environment: failed with initial frozen solve. 问题，如下图。

觉得情况不对，我手动ctrl+c取消了安装，查询了问题得到以下可能解决方案：

• 换源
• Python版本不对
• conda命令改用pip

换源命令如下

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

换过之后我还是卡在这里。

python版本，貌似pytorch-gpu版要求python为3.7-3.9，而我是python3.9所以也不是这个问题。

改用pip，在上面官网图中conda改为pip，其余不变，得到命令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

执行后成功安装，如下图。

验证

命令行中按顺序执行下列命令

python
import pytorch
torch.__version__
torch.cuda.is_available()
torch.cuda.device(0)
torch.cuda.device_count()
torch.cuda.get_device_name(0)

结果如下图

成功安装！