pytorch环境配置

2023-09-15 02:10:13

pytorch相关配置

参考视频链接:https://www.bilibili.com/video/BV1Ns4y1R7MP
我的系统:window11
GPU:4070laptop

一:安装cuda

1:找到nvidia control panel

在这里插入图片描述

2:打开控制面板,找到系统信息中的,组件即可查看到cuda版本,这里我的cuda版本为12.0.150

在这里插入图片描述

3:下载对应版本的 cuda toolkit(我下载的12.0.1)

下载网址:https://developer.nvidia.com/cuda-toolkit-archive

在这里插入图片描述
在这里插入图片描述

4:安装流程

此处建议使用默认路径
在这里插入图片描述
点击同意并继续
在这里插入图片描述
注意这里需要点击自定义安装,点击展开所有选项,如果出现新版本比当前版本低则取消打勾
在这里插入图片描述
此处可以自定义位置了:D:\ENV\cuda_manager\CUDA12.0.1\CUDA1
在这里插入图片描述
改好后点击下一步
在这里插入图片描述
此处勾选
在这里插入图片描述
点击下一步
在这里插入图片描述
点击关闭
在这里插入图片描述

5.验证是否安装成功

在anaconda prompt 窗口中输入 nvcc -V 查看版本
注意:确保是新打开的窗口,不然可能显示没有安装成功
在这里插入图片描述
我这里显示没有成功:
在这里插入图片描述
错误原因:nvcc -V v必须大写
在这里插入图片描述
cuda安装成功啦!!

二:配置cuDNN

1.版本

根据cuda版本去选
在这里插入图片描述

2.下载

网址:https://developer.nvidia.com/rdp/cudnn-archive
没有找到2023年1月的,就找了一个最相近的
在这里插入图片描述
点击后出现下拉菜单,直接点击下载即可
在这里插入图片描述
之前没有注册过的话,需进行注册
在这里插入图片描述

3.使用

下载完成解压过后,有三个文件夹lib,include,bin
在这里插入图片描述
将这三个文件夹复制到 D:\ENV\cuda_manager\CUDA12.0.1\CUDA1 中即可
在这里插入图片描述

三:安装pytorch

我的cuda是12.0 发现最新的pytorch是11.8,选择最新的即可
下面自动生成的命令可以用以配置相应的包
在这里插入图片描述

四:新建一个虚拟环境

要在 Jupyter Notebook 中为 PyTorch 创建一个虚拟环境,可以使用以下的步骤。这里,我会使用 `conda` 作为包管理器和环境管理器。如果你还没有安装 `conda`,可以考虑安装 Anaconda 或 Miniconda。
  1. 创建一个新的虚拟环境:

    打开命令行或终端,然后输入以下命令来创建一个名为 pytorch_env_cuda12_0 的新环境。你可以选择其他名称。

    conda create --name pytorch_env_cuda12_0 python=3.8
    

    选择 Python 3.8 作为该环境的版本,你也可以选择其他版本。

  2. 激活新创建的环境:

    conda activate pytorch_env_cuda12_0
    
  3. 安装 PyTorch:

    在已激活的环境中,根据你的 CUDA 版本和操作系统,使用适当的命令来安装 PyTorch。例如,对于 CUDA 12.0 (如果已经发布并支持):

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
    

    注意: 以上命令可能需要根据 PyTorch 的官方文档进行更改,以确保与 CUDA 12.0 兼容。

  4. 安装 Jupyter:

    在同一个环境中,安装 Jupyter:

    pip install jupyter
    
  5. 让 Jupyter Notebook 识别新环境:

    首先,确保 ipykernel 已安装:

    pip install ipykernel
    

    然后,将环境添加到 Jupyter:

    python -m ipykernel install --user --name=pytorch_env_cuda12_0 --display-name="pytorch_env_cuda12_0"
    
  6. 启动 Jupyter Notebook:

    你可以直接运行 jupyter notebook,然后在 Jupyter Notebook 中选择 “pytorch_env_cuda12_0” 作为运行内核。

现在,你应该已经在 Jupyter Notebook 中设置了一个专门的 PyTorch 环境。每次你想使用这个环境时,只需在 Notebook 中选择相应的内核即可。

五:测试

在这里插入图片描述
在这里插入图片描述
至此pytorch安装完成

更多推荐

观察者模式 & 发布-订阅模式(设计模式与开发实践 P8)

文章目录观察者模式运用实现观察者模式定义:他用来定义对象之间一种一对多的依赖关系,当一个对象状态发生改变时,所有依赖他的对象都会得到通知运用如果我们使用过DOM上的事件函数,那就接触过观察者模式document.body.addEventListener("click",function(){console.log("

HTML 知识扫盲

写在前面HTML是一门超文本标记语言,不管你听没听说过HTML,但在网上冲浪的途中你无时不刻都在与它接触,他遍布在每个你看得到的互联网的角落。其实对于笔者而言,我已经断断续续地学习过这门语言,经过时间的磋磨,所剩知识也是寥寥无几,这次借此机会复盘并总结一下HTML,当然在这里我不会将HTML语言的细节全盘拖出,只是总结

Android Fragment动画实现

在Android中,你可以使用FragmentTransaction来实现Fragment的动画效果。这允许你在添加、替换或移除Fragment时应用动画,从而改善用户体验。下面是如何实现Fragment动画的基本步骤:1.创建两个Fragment:首先,创建两个Fragment,例如FragmentA和Fragmen

TS同时打包和监视所有ts文件或只指定ts文件

当我们项目中ts文件较多时,我们如何直接打包所有ts文件为js文件?而不是使用tsc文件名一个一个去打包文件一、配置tsconfig.json文件创建一个tsconfig.json文件,该文件中不需要配置任何信息二、控制台输入打包命令在控制台输入如下代码:tsc三、对所有ts文件进行监听但是我们并没有对文件进行监听,修

Linux各种命令-查询篇

目录查看文件内容查看存储空间查看python安装目录查Ubuntu版本查看所有文件(含隐藏文件)查IP查看内存使用情况查看GPU使用情况查看CPU使用情况​​​​​​​查看文件内容cat[选项][文件...]-n:显示行号。-b:显示非空行号。-s:合并空白行。-E:在每行结尾添加$符号。-T:将制表符显示为^I。-v

交换机端口镜像详解

交换机端口镜像是一种网络监控技术,它允许将一个或多个交换机端口的网络流量复制并重定向到另一个端口上,以便进行流量监测、分析和记录。通过端口镜像,管理员可以实时查看特定端口上的流量,以进行网络故障排查、安全审计和性能优化。以下是关于交换机端口镜像的详细介绍:工作原理:交换机端口镜像通过在交换机的配置中指定源端口和目标端口

pycharm 中package, directory, sources root, resources root的区别

【遇到的问题】导入yolov5中有utils文件,自己的代码中也有utils文件,使得yolov5中的这部分引用出错了。【解决方案】单独建立detection文件夹,把检测相关的都放在这里,yolov5是github上拉取的源码,发现yolov5中fromutilsimport...有下划线,且会认为是edgeserv

多输入多输出 | MATLAB实现PSO-LSSVM粒子群优化最小二乘支持向量机多输入多输出

多输入多输出|MATLAB实现PSO-LSSVM粒子群优化最小二乘支持向量机多输入多输出目录多输入多输出|MATLAB实现PSO-LSSVM粒子群优化最小二乘支持向量机多输入多输出预测效果基本介绍程序设计往期精彩参考资料预测效果基本介绍MATLAB实现PSO-LSSVM粒子群优化最小二乘支持向量机多输入多输出1.dat

【码银送书第七期】七本考研书籍

八九月的朋友圈刮起了一股晒通知书潮,频频有大佬晒出“研究生入学通知书”,看着让人既羡慕又焦虑。果然应了那句老话——比你优秀的人,还比你努力。心里痒痒,想考研的技术人儿~别再犹豫了。小编咨询了一大波上岸的大佬,这份备考书单给大家参考。专业课(此处特指408,全称计算机专业基础综合)知识点超级多,题目灵活,数据结构、计算机

考研408 | 【计算机组成原理】 数据的表示和运算

进位计数制十进制计数法:推广:r进制计数法任意进制-->十进制:二进制<-->八进制、十六进制:各种进制的常见书写方式:十进制-->任意进制:十进制-->二进制(拼凑法):真值和机器数:总结:BCD码总结:无符号整数的表示和运算无符号整数在计算机中的应用:无符号整数的表示:无符号整数的加法运算:无符号整数的减法运算:总

uniapp运行到IOS真机提示 错误:请查看是否设备未加入到证书列表或者确认证书类型是否匹配

参考文章:请查看是否设备未加入到证书列表或者确认证书类型是否匹配ios开发描述文件必须绑定调试设备,只有授权的设备才可以直接安装基座,所以在申请开发描述文件之前,先添加调试的IOS设备。前往网站https://developer.apple.com,在Devices中,添加手机设备UUID第一步:登录第二步:检查设备列

热文推荐