当前位置: 首页 > article >正文

Windows10安装cuda11.3.0+cudnn8.5.0,以及创建conda虚拟环境(pytorch)

1、检查电脑驱动版本为561.09,选择cuda版本,下图可知cuda版本<12.6。

nvidia-smi  #查看驱动版本,以及最大可以安装的cuda版本

2、Anaconda3-2024.06-1-Windows-x86_64.exe下载:

官网:https://www.baidu.com/link?url=sGYwXBDoboSD5mXFY-MfdvEVhkNc8DK7jkyxRApBUIl4rvnSmzwuCYPwkNEy9T_HH458BpuRa3iRTttYRUI8ga&wd=&eqid=b5e8fd3f00001fc60000000466eb7f2b

清华镜像:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

后续具体步骤参照:学习记录:windows笔记本卸载cuda10.0、Anaconda3-2019、pycharm-2021,并跟新驱动重新安装cuda11.6、anaconda3-2022、pycharm-2022_nvidia cuda documention是否可以卸载-CSDN博客

3、cuda11.3.0下载:csdn - 安全中心

后续具体步骤参照:学习记录:windows笔记本卸载cuda10.0、Anaconda3-2019、pycharm-2021,并跟新驱动重新安装cuda11.6、anaconda3-2022、pycharm-2022_nvidia cuda documention是否可以卸载-CSDN博客

4、 cudnn8.5.0下载:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse850-116

       需登录NVIDIA账号,没有就用电子邮箱注册。

后续具体步骤参照:学习记录:windows笔记本卸载cuda10.0、Anaconda3-2019、pycharm-2021,并跟新驱动重新安装cuda11.6、anaconda3-2022、pycharm-2022_nvidia cuda documention是否可以卸载-CSDN博客

nvcc -V  #查看cuda版本
conda list #conda安装内容
pip list #pip安装内容

5、 创建一个名为<learn>的环境,python版本为3.9,需要注意的是pytorch与python对应。根据cuda11.3.0,参照下图我选择了pytorch=1.12.0、python=3.9。

conda create -n learn python==3.9
conda activate learn

#去网站找pytorch安装命令:https://pytorch.org/get-started/previous-versions/
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
#或者
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

 6、检查是否安装成功。

python
import torch
print(torch.__version__)  # pytorch版本
print(torch.version.cuda)  # cuda版本
print(torch.cuda.is_available())  # 查看cuda是否可用

http://www.kler.cn/a/315933.html

相关文章:

  • Linux 实现自动登陆远程机器
  • Wordpress常用配置,包括看板娘跨域等
  • 基于yolov8、yolov5的车型检测识别系统(含UI界面、训练好的模型、Python代码、数据集)
  • Android 项目依赖库无法找到的解决方案
  • 数字IC后端实现之Innovus specifyCellEdgeSpacing和ICC2 set_placement_spacing_rule的应用
  • Spring Cloud Gateway(分发请求)
  • sklearn特征选取之RFE
  • Django高级特性和最佳实践
  • SpringBoot 配置⽂件
  • 【数据结构与算法】LeetCode:二分查找
  • MATLAB给一段数据加宽频噪声的方法(随机噪声+带通滤波器)
  • 【Go】Go 环境下载与安装教程(Windows系统)
  • 九、成功版--windows上安装artifactory配置postgressql
  • [Redis][环境配置]详细讲解
  • Spark-累加器源码分析
  • JS执行机制(同步和异步)
  • 深度学习入门:探索神经网络、感知器与损失函数
  • html实现TAB选项卡切换
  • LLMs之OCR:llm_aided_ocr(基于LLM辅助的OCR项目)的简介、安装和使用方法、案例应用之详细攻略
  • Python之一些列表的练习题
  • Spring Boot入门:构建你的首个Spring Boot应用
  • Mybatis-plus进阶篇(二)
  • 【JUC并发编程系列】深入理解Java并发机制:线程局部变量的奥秘与最佳实践(五、ThreadLocal原理、对象之间的引用)
  • 数据结构 ——— 常见的时间复杂度计算例题(最终篇)
  • Linux驱动开发 ——架构体系
  • 求最大公约数