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

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

python 复制代码
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博客

python 复制代码
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。

python 复制代码
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 复制代码
python
import torch
print(torch.__version__)  # pytorch版本
print(torch.version.cuda)  # cuda版本
print(torch.cuda.is_available())  # 查看cuda是否可用
相关推荐
棒棒的皮皮4 分钟前
【OpenCV】Python图像处理之仿射变换
图像处理·python·opencv·计算机视觉
weixin_4462608513 分钟前
FastF1: 轻松获取和分析F1数据的Python包
开发语言·python
我送炭你添花15 分钟前
Pelco KBD300A 模拟器:06.用 PyQt5 实现 1:1 像素级完美复刻 Pelco KBD300A 键盘
python·qt·自动化·运维开发
山土成旧客16 分钟前
【Python学习打卡-Day22】启航Kaggle:从路径管理到独立项目研究的全方位指南
开发语言·python·学习
我命由我1234518 分钟前
Python Flask 开发 - Flask 路径参数类型(string、int、float、path、uuid)
服务器·开发语言·后端·python·flask·学习方法·python3.11
Chen--Xing31 分钟前
LeetCode LCR 119.最长连续序列
c++·python·算法·leetcode·rust
BoBoZz1935 分钟前
ConnectivityFilter数据集中分离的区域或连通分量
python·vtk·图形渲染·图形处理
程序员杰哥37 分钟前
如何使用Postman做接口自动化测试?
自动化测试·软件测试·python·测试工具·测试用例·接口测试·postman
最幸伏的人37 分钟前
Pycharm2025.3版本使用UV搭建项目
python·pycharm·uv
天才测试猿37 分钟前
Jmeter 命令行压测&生成HTML测试报告
自动化测试·软件测试·python·测试工具·jmeter·职场和发展·jenkins