本文主要参考PyTorch文档
不同操作系统,不同语言,不同包管理器安装PyTorch方法不一样。Anaconda是官网推荐的和主流的包管理器,若支持CUDA则能更好的用GPU进行加速,不过不是必选项。
首先查看是否支持GPU加速,即NVDIA显卡支持,这是硬件决定的,没有的憨憨不必去下NVDIA驱动软件了。
最直观的方法可以右键桌面,查看是否有NVDIA管理器,或者在cmd中输入nvidia-smi
查看NVDIA版本,很遗憾我的没有。
不死心的我右键此电脑->管理->设备管理器->显示适配器,实锤不是NVDIA,是AMD的。
如果有NAVID,进入NAVID控制面板就可以查看对应的CUDA版本;如果像我一样没有NAVID则不能使用CUDA和GPU加速,选择CPU,如下图。
即本文主要介绍基于Windows+Python+Anaconda的Pytorch环境搭建。无缘CUDA
Python环境搭建可以参考我这篇博客:Python环境搭建-从安装到Hello World
可以在Anaconda官网下载对应安装包(巨慢)。
或者在清华镜像中下载。
或者我也上传到了百度网盘, 提取码: f4ri
打开cmd,输入conda --version
即可查看Anaconda版本。
至此Anaconda安装成功( •̀ ω •́ )✧
(
插播反爬信息)博主CSDN地址:https://wzlodq.blog.csdn.net/
在cmd中执行,添加清华镜像源。官网的巨慢。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda config --set ssl_verify false
conda config --set show_channel_urls yes
创建一个虚拟环境,名称可以自定义,这里写的是myPytorch,Python版本可以在cmd中输入python --version
查看,比如我的是3.9.1
conda create -n myPytorch python=3.9.1
在cmd中输入activate base
,或者在菜单栏中打开Anaconda Prompt。
进入所创建的环境:conda activate myPytorch
第一节版本选择时,官网给出了相应安装命令,这里去掉-c pytorch
不从官网下载,在创建环境中执行以下命令即可:
conda install pytorch torchvision torchaudio cpuonly
在PyCham中添加创建的环境,点击File->settings->Project->Python Interpreter->Add Interpreter
import torch
print(torch.__version__)
原创不易,请勿转载(
本不富裕的访问量雪上加霜)
博主首页:https://wzlodq.blog.csdn.net/
来都来了,不评论两句吗
如果文章对你有帮助,记得一键三连❤