记录安装pytorch-gpu

摆渡阿狸

已于 2024-03-20 11:36:57 修改

阅读量925

点赞数 10

CC 4.0 BY-SA版权

文章标签： pytorch 人工智能 python

于 2024-01-19 16:32:12 首次发布

本文链接：https://blog.csdn.net/qq_15261081/article/details/135701257

本文详细指导了如何在特定CUDA版本（如11.8）下安装和配置PyTorch、torchvision、torchaudio，以及额外的库如cv2等，包括创建虚拟环境、下载对应版本的whl文件并进行安装，最后测试GPU版本的PyTorch是否成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 查看cuda版本

nvidia-smi

在这里插入图片描述
安装的cuda不得大于该版本

2. 安装torch、torchvision、torchaudio三个组件

2.1 下载 torch

下载网址
首先选择torch，ctrl + F 搜索 [cu118-cp38-cp38-win] 这里cu118 是下载的 CUDA 11.8 版本，cp38-cp38 是说 Python 版本是 3.8。
在这里插入图片描述

2.2 查看对应版本的torchvision、torchaudio

查看网址
ctrl + F 搜索 [pip install torch==2.0.0] 并且对应cuda为11.8。

在这里插入图片描述
因此torchvision需要安装0.15.1版本，torchaudio需要安装2.0.1版本。

2.3 下载torchvision、torchaudio

在之前的网址中选择 torchvision，ctrl + F 搜索 [cu118-cp38-cp38-win]，下载版本为 0.15.1 的。
在这里插入图片描述
选择 torchaudio，ctrl + F 搜索 [cu118-cp38-cp38-win]，下载版本为 2.0.1 的。

2.4 创建python虚拟环境并激活

conda create -n your_env_name python=x.x

conda activate your_env_name

2.5 安装

按照三个文件的存放路径进行安装，例如
在这里插入图片描述
对应的命令应该是

pip install C:\Users\sybs\Downloads\torch-2.0.0+cu118-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install C:\Users\sybs\Downloads\torchaudio-2.0.1+cu118-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install C:\Users\sybs\Downloads\torchvision-0.15.1+cu118-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

安装过程耐心等待。

安装结束后需要测试是否成功安装gpu版本的pytorch，这里依旧在创建的环境下输入python，进入python编程环境后输入 import torch 回车后输入 torch.cuda.is_available() 。如果返回True则安装成功。

其他

安装 cv2、tqdm、matplotlib、captum、skimage：

conda install --channel https://conda.anaconda.org/menpo opencv
conda install tqdm
conda install matplotlib
conda install captum
pip install scikit-image

删除环境：