小白从零开始配置pytorch环境

一、下载ANACONDA

官方网址Anaconda Installers and Packages

笔者选择的是Anaconda3-5.3.0-Windows-x86_64.exe版本。全程安装可以手机开热点,会快一点。

二、查看电脑是否有显卡

1、打开任务管理器

2、查看电脑CUBA版本,如上篇文章所提到查看CUDA-VERSION

笔者这里是11.3,安装cuda的时候要求小于等于11.3版本,最好是刚好等于,没有的话小于也可以。

三、创建环境

打开ANACONDA的命令行窗口,会显示在base环境里,新建一个环境名为pytorch(可自定义)。

复制代码
%%pytorch为环境的名字,可以自定义
conda create -n pytorch 

四、下载python3.6

在命令行窗口中输入

复制代码
conda create -n pytorch python=3.6

按提示输入y。

五、安装pytorch以及cuda

1、由base环境转入刚刚新建好的pytorch环境里

复制代码
conda activate pytorch

2、查看是否安装了torch已经torch的版本

复制代码
pip list

没安装的话则不会显示。

3、安装torch以及cu

在该网址download.pytorch.org/whl/torch_stable.html找到torch-1.3.0+cu92-cp36-cp36m-win_amd64.whl,下载到自定义路径。

笔者下载于E:\app\EDGE DOWNLOAD

输入下面代码,切换路径。

复制代码
cd E:\app\EDGE DOWNLOAD
E:

再输入下面代码安装torch与cu。

复制代码
pip install torch-1.3.0+cu92-cp36-cp36m-win_amd64.whl

4、查看是否下载成功

复制代码
pip list

六、验证是否安装成功

1、输入python会显示python版本信息,则代表Python安装成功

复制代码
python

2、输入下面代码验证,成功则返回ture。

复制代码
import torch
torch.cuda.is_available()

七、补充:torchvision下载

选择版本的网址:

download.pytorch.org/whl/torch_stable.html?spm=a2c6h.12873639.article-detail.9.49ae7f933ODqKL

参考上面torch的安装过程,用下面指令,改一下文件名即可。

复制代码
pip install torchvision-0.4.1+cu92-cp36-cp36m-win_amd64.whl

最终效果,pip list查看已下载文件。

相关推荐
奔跑吧邓邓子14 分钟前
DeepSeek 赋能智能教育知识图谱:从构建到应用的革命性突破
人工智能·知识图谱·应用·deepseek·智能教育
Mantanmu17 分钟前
Python训练day40
人工智能·python·机器学习
天天爱吃肉821820 分钟前
新能源汽车热管理核心技术解析:冬季续航提升40%的行业方案
android·python·嵌入式硬件·汽车
ss.li23 分钟前
TripGenie:畅游济南旅行规划助手:个人工作纪实(二十二)
javascript·人工智能·python
小天才才32 分钟前
前沿论文汇总(机器学习/深度学习/大模型/搜广推/自然语言处理)
人工智能·深度学习·机器学习·自然语言处理
l木本I36 分钟前
大模型低秩微调技术 LoRA 深度解析与实践
python·深度学习·自然语言处理·lstm·transformer
哆啦A梦的口袋呀40 分钟前
基于Python学习《Head First设计模式》第七章 适配器和外观模式
python·学习·设计模式
十月狐狸43 分钟前
Python字符串进化史:从青涩到成熟的蜕变
python
新加坡内哥谈技术1 小时前
Meta计划借助AI实现广告创作全自动化
运维·人工智能·自动化
狐凄2 小时前
Python实例题:Python计算线性代数
开发语言·python·线性代数