【超级简单】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

相关推荐
我是小哪吒2.02 分钟前
书籍推荐-《对抗机器学习:攻击面、防御机制与人工智能中的学习理论》
人工智能·深度学习·学习·机器学习·ai·语言模型·大模型
慕婉03076 分钟前
深度学习前置知识全面解析:从机器学习到深度学习的进阶之路
人工智能·深度学习·机器学习
阿蒙Amon11 分钟前
【Python小工具】使用 OpenCV 获取视频时长的详细指南
python·opencv·音视频
荔枝吻1 小时前
【AI总结】Git vs GitHub vs GitLab:深度解析三者联系与核心区别
人工智能·git·github
橘子编程1 小时前
Python-Word文档、PPT、PDF以及Pillow处理图像详解
开发语言·python
Jamie201901061 小时前
高档宠物食品对宠物的健康益处有哪些?
大数据·人工智能
蓝婷儿1 小时前
Python 机器学习核心入门与实战进阶 Day 2 - KNN(K-近邻算法)分类实战与调参
python·机器学习·近邻算法
云卓SKYDROID1 小时前
无人机载重模块技术要点分析
人工智能·无人机·科普·高科技·云卓科技
云卓SKYDROID1 小时前
无人机RTK技术要点与难点分析
人工智能·无人机·科普·高科技·云卓科技
之歆2 小时前
Python-封装和解构-set及操作-字典及操作-解析式生成器-内建函数迭代器-学习笔记
笔记·python·学习