基本信息:win10+64位,阅读此文章前建议先明确自己的电脑配置:右键我的电脑-属性,即可查看。
Pytorch安装之旅
一、Anaconda下载及安装
1.下载安装
-官网下载:https://www.anaconda.com/distribution/
-镜像下载(推荐):https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
安装部分不再赘述,按照步骤进行即可。
2.新建环境并激活
新建虚拟环境,按照我的理解来说,就像新建一个屋子。建好屋子后,再放入相应需要的物资(也就是我们下载的其他包)。
可以选择两种方法新建:
方法①:打开Anaconda Navigator,点击->Environments->点击左下角的Create
方法②:打开Anaconda prompt->输入conda create -n pytorch python=3.7->回车,
即建立完成。
激活方法有很多,这里就阐述一种:打开Anaconda prompt->输入activate pytorch->回车
即激活完成。(这里的pytorch是我新建的环境名,可以根据自己的需求对应修改)
补充:
—通过这两种方法新建的环境都是互通的,在Anaconda prompt新建的环境,打开Anaconda Navigator也能找到。
—要在哪个环境添加包,或是处理文件必须先激活该环境。
—相关查看信息的指令(在Anaconda prompt中输入):
查看当前的虚拟环境列表:conda info --envs
退出当前虚拟环境:deactivate 环境名
其他指令可以自行搜索学习
3.可能问题
——打开Anaconda Navigator一直处于loading状态?
——参考:https://blog.youkuaiyun.com/weixin_44821874/article/details/115281847
二、CUDA相关
1.查找相关配置信息
桌面右键,打开nvidia控制面板:
如图所示,对应的就是cuda11.1。
2.相关下载及配置
点击链接下载相应版本:https://developer.nvidia.com/cuda-toolkit-archive
图中对应选项为win10+64位+本地下载。点击右下角下载安装即可。安装可按照默认选项进行。
3.安装结果确认
win+R,输入cmd回车,即打开命令提示符:输入nvcc -V
正常返回版本即为安装成功。
4.可能问题
——流程正确,仍不能返回版本?
——关闭cmd,重新打开。
若仍存在错误,可能配置出现错误,需要修改环境变量,修改后重启cmd。参考:https://blog.youkuaiyun.com/weixin_45731793/article/details/107975494?utm_medium=distribute.pc_relevant.none-task-blog-2defaultbaidujs_title~default-4.no_search_link&spm=1001.2101.3001.4242
三、Pytorch下载
注意,这里安装需要激活相应的虚拟环境。
1.安装方法一
官网下载:https://pytorch.org/
选择相应配置后,复制相应指令,打开Anaconda prompt输入,回车即可下载。
该方法速度较慢,容易出错,不太推荐。
2.安装方法二(推荐)
下载:
https://download.pytorch.org/whl/torch_stable.html
找到相应版本下载并记住保存地址。
图中所示代表:cuda11.1+torchvision0.10.0+python3.7+Windows+64位
其他以此类推。
想要正常运行,一般需要下载pytorch+torchvision。
下载后,本地安装:
相应环境中输入:pip install 对应whl文件的路径
pytorch+torchvision都按照此方法安装即可。
3.安装结果确认
依次输入指令,若正常运行即为安装成功。
import torch
print(torch.__version__)
print(torch.cuda.is_available())
4.可能问题
——采用第一个方法安装后,pytorch未在相应虚拟环境中?
——直接采用第二种方法重新安装即可。(这个我也没想清楚哈哈)
——版本选择?
——显卡有对应的版本,最好下载相应版本的cuda,其余情况不是很清楚。
四、Pycharm
1.配置环境
新建项目后:
点击2->add
找到相应虚拟环境里的python.exe,点击OK即可。
2.结果确认
运行程序,得到正确结果即为成功。可以开始玩耍啦~