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是否可用
相关推荐
zhangbin_2371 分钟前
【Python机器学习】NLP信息提取——正则模式
开发语言·人工智能·python·深度学习·机器学习·自然语言处理
小李飞刀李寻欢5 分钟前
mongoDB 读取数据python版本实现
数据库·python·mongodb·数据
大脸猫吖13 分钟前
Selenium4.0实现自动搜索功能
python·selenium·自动化
你可以自己看17 分钟前
python中数据处理库,机器学习库以及自动化与爬虫
python·机器学习·自动化
kolaseen36 分钟前
pytorch的动态计算图机制
人工智能·pytorch·python·深度学习·机器学习
约束11239 分钟前
ipynb脚本是运行在本地还是云端?
python·jupyter
QXH2000001 小时前
数据预处理方法—数据标准化和数据归一化
人工智能·python·深度学习·机器学习
5pace1 小时前
动手学深度学习(李沐)PyTorch 第 2 章 预备知识
人工智能·pytorch·深度学习
一丝晨光1 小时前
语言的循环语句
java·c++·python·c#·c·fortran·algol
NuyoahC1 小时前
Python语法(二)——函数
开发语言·笔记·python