GPU版Pytorch的安装

下载Pycharm

略

下载Miniconda

Distribution Installers和Miniconda Installers的区别：
简单来说，​Anaconda Distribution Installers 是一个“全家桶”，而 Miniconda Installers 是一个“基础版”​。Miniconda一个最小的安装包。只包含Conda、Python以及少量核心依赖包​（如pip, zlib等）。​不预装任何科学计算库。

启动conda和虚拟环境

启动输入：

conda info

创建虚拟环境输入：（可以改成自己的名字和想要的python版本）

conda create -n pollyhub_3_9 python=3.9

启动虚拟环境输入：

conda activate pollyhub_3_9

下载对应的Pytorch版本

在cmd输入；

nvidia-smi

找到可以适配的最新的cuda版本

将最后的数字改成对应的版本：

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

测试是否成功：

import torch
print(torch.__version__)          # 应显示类似 2.3.0+cu121
print(torch.cuda.is_available())  # 应返回 True
print(torch.cuda.device_count())  # 应返回 ≥1
print(torch.cuda.get_device_name(0))  # 应显示你的 GPU 型号（如 RTX 3060）