【简易 教程:Pytorch 配置 GPU版本】
创始人
2024-04-04 17:13:24
0

Pytorch 配置 GPU 版本

博主 是 通过 Anaconda 来管理配置Python工具包的。至于Anaconda 的下载安装,比较简单,在其官网下载,基本就是一步接着一步就好~ 这里我不记录咯 ~如果大家有需求的话,留言就好 ~

------>>首先,通过 终端进入conda 界面:
在这里插入图片描述
------>>其次,为了在安装过程中速度快一点点,我们先设置下镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
conda config --set show_channel_urls yes

需要指出的是,上述每一行命令,都是需要每次执行一句哈,即 你复制时,复制一行,就执行一行,(而不是把这些全部复制了,再全部粘贴!这样不行的哦!!)

另外,如果有朋友之前已经引入过镜像源了,那么会报一些 Warning,这无关紧要,请继续执行命令即可。

------>> 为了便于Pytorch的安装,我们为它单独创立一个存放其环境的地方:conda create -n pytorch python=3.9,后面的3.9指的是Python的版本,这个可以根据需要自己选择。

在输入上述命令过后,我们应该会看到如下界面:
在这里插入图片描述

------>>紧接着,我们切换环境,进入刚刚建立的Pytorch:conda activate pytorch,并且输入命令下载 符合我们自己电脑版本的 GPU版的Pytorch:conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia,注意,该命令不是随意输入的,需要根据自己的电脑 里面NVIDIA的版本。

点击 我 可以 查看符合自己版本的下载语句!!!

进去pytorch官网,下拉,可以看到如下所示的界面:(注意要 选择 和自己版本匹配的喔!!!这样就可以看到对应的语句了~)

在这里插入图片描述

为了下载快一点,可以把上述语句的 -c python 去掉。即 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia 这样就好。这只是我的一个示例,具体的可以根据自己选择来。
在这里插入图片描述

------>>然后!就是 漫长的安装界面了,期间 可能还需要你 输入 y并且按回车确定,这个知道就行,操作还是比较简单的。

就比如这样:(漫 长 的 等 待 --)
在这里插入图片描述
安装完成后。博主再继续记录(更于 2022.11.6 21:59)

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...