安装GPU版本的pytorch

一、在 pytorch官网Get Started界面选择合适的版本，官网会给出相应的语句

安装之后发现结果是CPU版本，重试几次之后结果还是一样

二、解决方案

可以参考博主windSeS的方法，在命令行中指定[python|cudatoolkit|pytorch]这三个版本不同的组合

但是在确定镜像源中存在我所指定的组合版本后，下载的torch仍然是CPU版本

如果出现这种情况，可以尝试使用pip安装

在pytorch官网previous pytorch version界面下找到自己cuda版本对应的pip安装命令行

我的cuda版本为11.1，选择安装低于我目前版本的10.2

复制其中#CUDA10.2下的安装命令

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

最后验证是否安装成功

>>> import torch
>>> torch.cuda.is_available()
True

安装好啦

总结：还是多尝试官网给出的方案，比较简单可行。如安装中遇到问题，可以选择安装较低版本的pytorch。

安装GPU版本的pytorch相关推荐

安装GPU版本的pytorch（解决pytorch安装时默认安装CPU版本的问题）保姆级教程
1.安装anaconda anaconda官网:www.anaconda.com 2.修改下载源为清华源由于pytorch的服务器在国外,直连下载的话很慢,所以选用清华镜像源下载 # 添加清华镜像 ...
win10/11下wsl2安装gpu版的pytorch（避坑指南）
0x00 注意不想折腾的不要弄了,老老实实用windows,现在WSL坑还很多. 想安装的一定要看官方文档!!在文末本教程只说明在安装了 WSL2 后,并且默认系统是 win11 下安装中的一些坑 ...
全网最详细之如何安装gpu版的pytorch
安装gpu版pytorch 一.准备: 1.需要查看nvidia驱动是否安装安装驱动可参考:https://blog.csdn.net/qq_41428418/article/details/119 ...
安装CPU版本的pytorch
题外话:有些同学的笔记本电脑可能没有显卡或者显卡不太好,建议安装CPU版本的pytorch.这类同学如果实在需要使用GPU加速训练过程,建议使用在线GPU. 打开Anaconda Prompt 查看环 ...
tensorflow运行环境linux,在ubuntu或者min运行环境下安装gpu版本的tensorflow
转载请注明出处:blog.csdn.net/sproll 本文描述在ubuntu系列操作系统上安装gpu版本的tensorflow的过程. 0,准备工作 BIOS中关闭板载显卡,显示器接在nvidia ...
英伟达3060Ti安装GPU版本TensorFlow2.X
查看Python与TensorFlow对应版本安装GPU版本的TensorFlow的时候,我们需要考虑的一个问题是Python版本与TensorFlow版本的对应关系,可以参考下面这个链接: Pyt ...
windows10 安装gpu版本TensorFlow脑壳疼自我记录
写文章时间为:2018.10.26 显卡为NVIDIA GeForce RTX 2080Ti(刚出没有多久),系统为window10 放在开始想说的是,以下是我安装过程,仅是自我总结,仅供参考. 再之 ...
常识——windows的tensorflow安装gpu版本,cuda算力只有3.0也可以
文章目录问题 windows安装tensorflow_gpu版本检查显卡是否支持cuda 查看自己的显卡型号去官网查看是否支持cuda 检查显卡的cuda算力按照GPU算力3.0的电脑配置进行 ...
安装老版本的Pytorch
参考安装老版本的Pytorch - 云+社区 - 腾讯云 Installing previous versions of PyTorch We'd prefer you install the lat ...

安装GPU版本的pytorch

安装GPU版本的pytorch相关推荐

最新文章

热门文章