如何安装gpu版本的pytorch(超详细)

该博客主要介绍了PyTorch的安装步骤。首先需检验是否已安装CUDA,若未安装可参考另一篇文章。接着要检查电脑的GPU与CPU、查询Python版本。然后进入官方页面,根据版本找到对应网址,下载whl文件并复制地址到Python配置环境位置,按顺序pip安装。最后检验安装是否成功。
该文章已生成可运行项目,

目录

1.安装CUDA

2.安装pytorch

1、检查自己电脑的gpu与cpu

2.查询你的python版本

3.安装pytorch

1、进入官方页面pytorch

2、pytorch的whl下载地址

3.检验pytorch安装是否完成


1.安装CUDA

首先,在命令行输入nvcc -V检验是否已安装CUDA

若结果显示为

则表明已经配置过CUDA。

若结果为

则尚未配置,配置过程可见笔者的另一篇文章CUDA安装教程(超详细Windows版)

2.安装pytorch

1、检查自己电脑的gpu与cpu

在左下角搜索栏中输入设备管理器,从中找到显示适配器即可看到电脑里的gpu与cpu了。

这里就可以看出你是否有gpu了。

2.查询你的python版本

ctrl+r后输入python

3.安装pytorch

1、进入官方页面pytorch

找到这个页面

从中根据我们的 版本在Run this Command中找到其所对应的网址

https://download.pytorch.org/whl/cu117从中找到我们所要安装的torch torchvision torchaudio

会相对复杂一点。

2、pytorch的whl下载地址

直接进入这个页面开始查找

这里cpu是安装cpu版本的cu是gpu版本的,cp指的是你的python版本,还有就是win和linux系统的选择,找到你所对应的版本将其点击下载。

找到这三个文件将其地址复制到你python的配置环境位置。

在这里进入命令窗口或者cd到这里

对这三个文件分别pip安装,

要按照torch,torchvision,torchaudio的顺序安装完成。

3.检验pytorch安装是否完成

如果结果为True则安装成功。

​​​​​​​​​​​

本文章已经生成可运行项目
### 安装GPU版本PyTorch 对于希望利用GPU加速计算的应用场景,在特定操作系统上安装GPU版本PyTorch是一个重要的步骤。考虑到不同的硬件和软件环境,以下是针对不同情况下的指导。 #### Ubuntu 18.04上的安装指南 在Ubuntu 18.04环境下准备安装GPU版本PyTorch时,首先要确认系统的显卡能够被CUDA所支持[^3]。例如,如果使用的显卡为GT640m,则需要注意该款显卡可能无法兼容最新的CUDA版本如11.0;相反,它更适合搭配较低版本的CUDA,像CUDA 10.0这样的选项可能会更加合适。 为了确保顺利安装,建议先访问[NVIDIA开发者页面](https://developer.nvidia.com/cuda-gpus),查询具体的显卡型号是否列于官方支持列表内,并据此选择合适的CUDA版本进行下载与安装[^4]。 完成上述准备工作之后,可以通过Conda或pip工具来获取指定版本PyTorch。这里推荐使用Anaconda管理Python包及其依赖关系,因为这能简化跨平台开发过程中的许多复杂度。具体操作命令如下所示: ```bash # 创建一个新的虚拟环境(可选) conda create -n pytorch_env python=3.7 conda activate pytorch_env # 安装PyTorch (假设选择了CUDA 10.0) conda install pytorch torchvision torchaudio cudatoolkit=10.0 -c pytorch ``` 以上命令会自动处理好所有必要的依赖项,包括但不限于CUDA Toolkit以及cuDNN库等。值得注意的是,当从官方网站挑选旧版PyTorch时,应当留意区分其分发形式——即是从`conda`渠道还是通过`.whl`文件安装,这一点同样适用于其他发行方式的选择[^5]。 最后一步是验证安装成功与否。可以在Python解释器中运行以下代码片段测试是否存在可用的GPU资源供PyTorch调用: ```python import torch print("GPU available:", torch.cuda.is_available()) ``` 这段简单的脚本将会输出当前环境中是否有有效的GPU设备可以用于加速运算。如果有多个GPU存在于同一台机器上,还可以进一步探索如何设置默认使用的GPU编号等功能特性。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值