可恶的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)
结果如下图
成功安装!