安装能调用GPU的pytorch版本

晨尘光

于 2024-03-01 14:21:10 发布

阅读量220

点赞数 2

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/DustDawn/article/details/136394896

版权

一般需要安装的库的版本和本机安装的CUDA版本保持一直才可以用，所以先查看自己装的CUDA版本
在终端（windows下的CMD窗口或PowerShell窗口）中输入命令

nvcc --version

以cuda10.2为例

# CUDA 10.2
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html

该命令是一个用于安装特定版本的 PyTorch 和 torchvision 的 pip 命令。这里我们详细解析一下这个命令的各个部分：

pip install: 这是 Python 的包管理工具 pip 的命令，用于安装 Python 包。
torch==1.9.0+cu102: 这部分表示要安装的 PyTorch 的版本。torch 是 PyTorch的主要包，==1.9.0
表示要安装的版本号，而 +cu102 表示这个 PyTorch 版本是为 CUDA 10.2 编译的。CUDA是 NVIDIA的并行计算平台和 API 模型，允许使用 GPU 进行通用计算。
torchvision==0.10.0+cu102: 这部分表示要安装的 torchvision 的版本。torchvision
是PyTorch 的一个包，包含了许多用于计算机视觉的模型、数据集和转换函数。==0.10.0 表示要安装的 torchvision的版本号，+cu102 同样表示这个 torchvision 版本是为 CUDA 10.2 编译的。
-f https://download.pytorch.org/whl/torch_stable.html: 这部分指定了一个 -f 或 --find-links 参数，用于指定一个查找链接（在这里是一个 HTML 页面）。pip将会从这个页面下载并安装指定的包。这通常用于从非 PyPI 源（如官方 PyTorch 网站）安装包。

总的来说，这个命令的目的是从 PyTorch 官方网站下载并安装特定版本（1.9.0 对于 PyTorch，0.10.0 对于 torchvision）且为 CUDA 10.2 编译的 PyTorch 和 torchvision。这通常用于确保安装的 PyTorch 版本与你的 CUDA 版本兼容。