【超级简单】3步 安装conda + pytorch gpu版本

【超级简单】3步 安装conda + pytorch gpu版本

  • [1. 创建虚拟环境](#1. 创建虚拟环境)
    • [至于python 3.X和cuda对应 ?](#至于python 3.X和cuda对应 ?)
  • [2. 下载cuda 驱动,cuda toolkit (可选)](#2. 下载cuda 驱动,cuda toolkit (可选))
  • [3. 虚拟环境中输入以下指令](#3. 虚拟环境中输入以下指令)
    • [至于 怎么查看自己电脑对应的cuda版本号 ?](#至于 怎么查看自己电脑对应的cuda版本号 ?)
  • 4.测试是否安装成功

下载anaconda,能让你高效管理不同版本运行环境。

conda + pytorch gpu

适用于linux 和window版本,

建议前提确保自己的显卡能用哈。

以下是在window10的示范。

1. 创建虚拟环境

conda create -n your_env_name python=3.X

conda avtivate your_env_name

至于python 3.X和cuda对应 ?

参考Tensorflow 的资料:

https://www.tensorflow.org/install/source_windows?hl=zh-cn

2. 下载cuda 驱动,cuda toolkit (可选)

https://developer.nvidia.com/cuda-toolkit-archive

下载成功的话可以 在cmd 输入 nvcc -V, 显示版本号。

显示 not found 则说明没转成功。

3. 虚拟环境中输入以下指令

复制代码
conda install pytorch torchvision torchaudio cudatoolkit=X.X  -c pytorch

X.X替换成你的版本号

至于 怎么查看自己电脑对应的cuda版本号 ?

:鼠标右键------>NVDIA控制面板------>帮助------>系统信息------>组件,product name 那一列就是nvdia对应版本。

4.测试是否安装成功

bash 复制代码
torch.cuda.is_avaliable()
# True, gpu is ok
# false, try to fix it by rebooting up. 

if not, there are some pro. with your settings, like conda environment doesnt correct, cuda Ver. is not right.

有疑惑可以再看看这篇文章写的。

笔墨多一点。

具体可参考:https://blog.csdn.net/qq_36851515/article/details/107608967

相关推荐
幂简集成17 分钟前
Realtime API 语音代理端到端接入全流程教程(含 Demo,延迟 280ms)
人工智能·个人开发
龙腾-虎跃24 分钟前
FreeSWITCH FunASR语音识别模块
人工智能·语音识别·xcode
智慧地球(AI·Earth)28 分钟前
给AI配一台手机+电脑?智谱AutoGLM上线!
人工智能·智能手机·电脑
Godspeed Zhao28 分钟前
自动驾驶中的传感器技术46——Radar(7)
人工智能·机器学习·自动驾驶
学生信的大叔40 分钟前
【Python自动化】Ubuntu24.04配置Selenium并测试
python·selenium·自动化
limengshi13839244 分钟前
机器学习面试:请介绍几种常用的学习率衰减方式
人工智能·学习·机器学习
AKAMAI1 小时前
Sport Network 凭借 Akamai 实现卓越成就
人工智能·云原生·云计算
周末程序猿2 小时前
机器学习|大模型为什么会出现"幻觉"?
人工智能
JoannaJuanCV2 小时前
大语言模型基石:Transformer
人工智能·语言模型·transformer
飞哥数智坊2 小时前
Qoder vs CodeBuddy,刚起步就收费,值吗?
人工智能·ai编程