【PyTorch】环境配置

Window11+RTX4060配置CUDA11.8+pytorch2.0.0

下载CUDA11.8

进入官网下载Link,然后点击Archive of Previous CUDA Releases,找到对应版本11.8。

选择对应操作系统版本,点Download下载到本地。

下载cuDNN

点击链接Link,找到CUDA11.8对应的cuDNN。

安装CUDA11.8

全程点击next直到安装结束。

配置环境变量,在path里添加。

bash 复制代码
#前两个应该自动配好了
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib

在命令行窗口输入以下命令看是否成功

bash 复制代码
nvcc -V

成功显示以下内容表示安装成功

安装cuDNN

解压下载的压缩包

把里边的三个文件夹bin,include,lib复制到下面路径(如果改了CUDA的安装路径,就复制到对应文件夹里)

复制代码
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

下载并安装python

官网链接

下载并安装Anaconda

官网链接选择适合自己系统的 64 位安装包,注意选择 Python3 以上的版本。

安装torch

进入官网Link,找到v2.0.0,用pip语句安装。用conda可能会出错,具体见安装pytorch报错torch.cuda.is_available()=false的解决方法

bash 复制代码
# CUDA 11.8
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

参考链接

win11+4060配置CUDA11.8+pytorch2.0.0

相关推荐
Robot2513 分钟前
「地平线」创始人余凯:自动驾驶尚未成熟,人形机器人更无从谈起
人工智能·科技·机器学习·机器人·自动驾驶
深蓝学院7 分钟前
开源|上海AILab:自动驾驶仿真平台LimSim Series,兼容端到端/知识驱动/模块化技术路线
人工智能·机器学习·自动驾驶
一点.点7 分钟前
LLM应用于自动驾驶方向相关论文整理(大模型在自动驾驶方向的相关研究)
人工智能·深度学习·机器学习·语言模型·自动驾驶·端到端大模型
geovindu8 分钟前
PyCharm 2023升级2024 版本
ide·python·pycharm
云天徽上18 分钟前
【数据可视化-41】15年NVDA, AAPL, MSFT, GOOGL & AMZ股票数据集可视化分析
人工智能·机器学习·信息可视化·数据挖掘·数据分析
EasyDSS19 分钟前
EasyCVR视频汇聚平台助力大型生产监控项目摄像机选型与应用
网络·人工智能·音视频
qq_4369621826 分钟前
奥威BI+AI数据分析解决方案
人工智能·数据挖掘·数据分析·ai数据分析
freexyn33 分钟前
Matlab自学笔记五十二:变量名称:检查变量名称是否存在或是否与关键字冲突
人工智能·笔记·算法·matlab
roc-ever36 分钟前
用Python做有趣的AI项目5:AI 画画机器人(图像风格迁移)
人工智能·python·深度学习
说私域38 分钟前
从大众传媒到数字生态:开源AI智能名片链动2+1模式S2B2C商城小程序驱动的营销革命
人工智能·小程序·开源·零售