pytorch-gpu安装的经验与教训
作者：故y 发布时间：2022-01-11 20:23:36
标签：pytorch-gpu,安装
首先说明
本文并不是安装教程，网上有很多，这里只是自己遇到的一些问题
我是以前安装的tensorflow-gpu的，但是发现现在的学术论文大部分都是用pytorch复现的，因此才去安装的pytorch-gpu
查看自己安装的CUDA
nvcc -V
这里我提供一个安装tensorflow时所用的CUDA对应表
安装cuDNN时版本一定要对应正确
安装完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的信息全部删除，再按照前面的流程走一遍，相信你自己一定会出现
来源：https://www.cnblogs.com/kk-style/p/17046809.html
0
投稿猜你喜欢
以下备忘升级至 Vue CLI 3.x 版本后，将项目目录改为新结构时所需做的一些改动。1. 卸载与安装npm uninstall vue-- 今天尝试了一下据说最好用的编译器Jupyter Lab，安装期间遇到了很多问题，以此为记录。1.安装jupyter labjupyter l
- 一、概述spark 有三大引擎，spark core、sparkSQL、sparkStreaming，spark core 的关键抽象是 S
- 简介A的独有+AB的公有B的独有+AB的公有AB的公有A的独有B的独有A的独有+B的独有+AB的公有A的独有+B的独有练习建表部门表DROP
- 条件选取：torch.where(condition, x, y) → Tensor返回从 x 或 y 中选择元素的张量，取决于 condi
- numpy的np.fromfile会出现如下的问题，只能一次性读取文件的内容，不能追加读取，连续两次的np.fromfile读到的东西一样如
- 前言大家都知道，早期的XMLHttpRequest不支持文件上传，一般用第三方js插件或者flash，现在可以借助XMLHttpReques
- 本文实例讲述了Yii开启片段缓存的方法。分享给大家供大家参考，具体如下：1、main.php文件中的components中添加：cache&
- 在mysql数据库开发中，我们有时候需要复制或拷贝一张表结构和数据到例外一张表，这个时候我们可以使用create ... select ..
- PyTorch基础入门三：PyTorch搭建多项式回归模型 1）理论简介对于一般的线性回归模型，由于该函数拟合出来的是一条直线，所
- 而由于InnoDB是一个健壮的事务型存储引擎，已经有10多年的历史，一些重量级的互联网公司(Yahoo,Google Netease ，Ta
- Microsoft SQL Server Management Studio是SQL SERVER的客户端工具，相信大家都知道。我不知道大伙
- 什么是虚拟环境这是 Python 3.3 的新特性：https://www.python.org/dev/peps/pep-0405/假设自
- 当然有其它工具可以做这件事，但如果客户不允许你在服务器乱装东西时这个脚本就会有用了。 代码如下:DECLARE @tbImportTable
- 如下所示：#!/usr/bin/python# coding=utf-8#import timeimport datetime# 今天日期t
- 这篇文章主要介绍了通过实例了解Python str()和repr()的区别,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参
- 对于yolo系列，应用广泛，在win10端也有很大的应用需求，所以这篇文章给出win10环境下的安装教程。先给出系列文章win10+anac
- 摘要：本文主要是在pandas中如何对字符串进行切分。我们考虑一下下面的应用场景。这个是我们的数据集(data)，可以看到，数据集中某一列(
- 开门见山自动化测试过程中，一般测试结果都会以邮件的形式发送给相关人员，那么，在Python中，如何编写代码将邮件发送给对应的用户？同时，发送
- 前言mysql 相信大部分人都用过，索引肯定也是用过的，但是你知道如何创建恰当的索引吗？在数据量小的时候，不合适的索引对性能并不会有太大的影
