首页 > 其他分享 >安装pytorch-gpu的经验与教训

安装pytorch-gpu的经验与教训

时间:2023-01-12 15:35:10浏览次数:65  
标签:教训 torch pytorch cuda print gpu 安装

首先说明

本文并不是安装教程,网上有很多,这里只是自己遇到的一些问题

我是以前安装的tensorflow-gpu的,但是发现现在的学术论文大部分都是用pytorch复现的,因此才去安装的pytorch-gpu

查看自己安装的CUDA

nvcc -V

这里我提供一个安装tensorflow时所用的CUDA对应表

 

 

 安装完CUDA时要改一下环境变量

 

 只用添加这个就好,其他三个CUDA会自动为你添加

 我的是cuda-11.2

但是官网没有配套的,直接就挑自己喜欢的下就行Start Locally | PyTorch

 

 这里强调:千万不要相信在其他网站上的下载命令。有可能你下载的就是cpu版本

安装的时候可能会遇到

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

不要担心,不用管它,耐心等一等,就会出现(y/n)选择y就可以了

这里要看一眼你的pytorch是不是cuda版本的,不是要重新下载

测试安装是否成功

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

当然你这些都做完了,还是出现了False

失败后的操作

打开你的anaconda

 

 将这里关于pytorch的信息全部删除,再按照前面的流程走一遍,相信你自己一定会出现

 

标签:教训,torch,pytorch,cuda,print,gpu,安装
From: https://www.cnblogs.com/kk-style/p/17046809.html

相关文章

  • PyTorch分布式并行训练
    目录Pytorch分布式并行训练原理分布式数据并行原理PyTorch分布式数据并行DemoPyTorch分布式数据并行内部设计PyTorch分布式数据并行源代码分析RingAllReduce算法参考文献......
  • PyTorch如何加速数据并行训练?分布式秘籍大揭秘
    PyTorch在学术圈里已经成为最为流行的深度学习框架,如何在使用PyTorch时实现高效的并行化?在芯片性能提升有限的今天,分布式训练成为了应对超大规模数据集和模型的主要......
  • 关于使用gpu解码图片的一些想法
    最近在使用gpu解码数据,数据是base64编码的图片,或者是二进制编码的图片以前都是使用cpu进行解码,也导致cpu使用率比较高使用gpu解码以后存在的问题就是:以yolov5为例,使用pgu......
  • Understanding dimensions in PyTorch
     在二维的操作实例以及输出的结果>>x=torch.tensor([[1,2,3],[4,5,6]])>>x.shapetorch.Size([2,3])>>torch.sum(x,dim=0)tenso......
  • 【Deep Learning Based On Pytorch】张量
    Pytorch中张量与Tensor()一、张量Tensor张量是一个统称,其中包含很多类型:0阶张量:标量、常数,0-DTensor1阶张量:向量,1-DTensor2阶张量:矩阵,2-DTensor3阶张量........
  • pytorch geometric并行运算要注意的一些问题
    如果有多gpu的情况,可以这样写:fromtorch_geometric.nnimportDataParalleliftorch.cuda.device_count()>1:model=DataParallel(model,device_ids=[0,1])这......
  • pytorch 损失函数
    损失函数在深度学习领域是用来计算搭建模型预测的输出值和真实值之间的误差。具体实现过程:在一个批次(batch)前向传播完成后,得到预测值,然后损失函数计算出预测值和真实值之间......
  • Pytorch GPU 环境搭建
    之前⼀直使⽤Tensorflow训练模型,第⼀次训练Pytorch模型的时候,发现速度很慢,仔细观察,发现GPU内存占⽤为0,基本没有使⽤GPU。AssertionError:CUDAunavailable,invaliddev......
  • GPU笔记
    图形流水线内存里面专门开了一块空间用来存储即将显示到显示器上的【像素】们。用你家显示器分辨率乘积的结果就可以算出你家的显示器可以显示多少个像素。每个像素用RGB2......
  • YOLOv5多GPU训练
    一、准备安装yolov5依赖,要求Python>=3.8以及PyTorch>=1.7gitclonehttps://github.com/ultralytics/yolov5#clonerepocdyolov5pipinstall-rrequirements.txt123二......