安装GPU版本的PyTorch

一、安装Anaconda

1、创建虚拟环境（anaconda编辑器）

conda create -n ywqpytorch python=3.9

2、进入虚拟环境

conda activate ywqpytorch

conda list #看有哪些包
conda env list #看有哪些虚拟环境
conda remove -n ywqpytorch --all #删除虚拟环境

二、GPU版本 配置CUDA

1、确定自己显卡算力 

（我的GPU）

2、确定自己可选的CUDA Runtime Version 

CUDA11.8

3、确保自己的CUDA Driver 版本>=CUDA Runtime 版本

nvidia-smi 查看CUDA Driver版本为12.1 

4、更新驱动

网址：nvidia.cn 安装显卡驱动最新版本 12.6

安装cuda位置

nvcc --version  判断cuda是否安装成功

5、打开命令窗口查看nvidia-smi 确定版本为12.6

我的12.6>11.8

三、用CUDA或pip安装pytorch

一起安装了pytorch、torchvision、torchaudio

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

注意：以下情况才是安装的GPU版本

四、用CUDA安装pytorch失败 选择pip

速度太慢 将torch的下载地址 https://download.pytorch.org/whl/cu118/torch-2.4.0%2Bcu118-cp39-cp39-win_amd64.whl 加入到迅雷中下载，下载成功将下载好的文件拖入

C:\Users\于宛晴>pip install E:\迅雷下载\torch-2.4.0+cu118-cp39-cp39-win_amd64.whl

失败报错

换镜像重新下载：

pip install xxxx -i https://pypi.tuna.tsinghua.edu.cn/simple

下载成功！！！

接着下载torchaudio和torchvision

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

判断是否安装成功

五、pycharm下载

新建项目 ——点击add Interpreter——选择anaconda安装目录，script文件夹，选择conda.exe文件，然后load就可以了，在下边出现的下拉菜单里边选择自己创建的虚拟环境。

import torch
print(torch.cuda.is_available())

输出True 安装成功