pytorch环境配置(pycharm)
1.首先安装anaconda
官网上面一般是最新版本的(目前是python3.9)
但是很多包不支持新版本的python,如果需要旧版本的,那可以到这里来下载https://repo.anaconda.com/archive/
其中Anaconda2/3表示适配的是python2.X还是3.X,后面的数字表示的是Anaconda的版本号,基本上百度一下版本号就能查到对应的是python的版本
下载好以后安装
记得跳过VScode
安装最后一页两个勾都不勾选就行
然后我们来检查一下是否安装成功
直接开始菜单找一下anaconda prompt
如果左边是一个base就安装成功了
如果想要安装gpu版本的pytorch,需要看一下显卡驱动是否正确安装
打开任务管理器(cmd输入taskmgr)
正常显示型号就表示安装正常,不然需要在英伟达官网下载一下对应的驱动
打开anaconda prompt
输入安装pytorch的命令conda create -n XXX(随便写个名字例如pytorch) python=X.X(对应的python版本号)
输入y确定
后面如果需要改变环境,可以用 conda activate XXX(刚才起的环境名)命令
如果想看这个环境里面有什么包,可以直接pip list
就会有这些包。
然后我们讲一下pytorch的安装
下拉就有这些对应版本安装的命令,如果你需要下载GPU版本的,就根据自己的显卡CUDA版本去下载,没有就直接选择CPU版本
CUDA版本可以直接在cmd里面输入nvidia-smi
就可以看到我的CUDA版本就是11.5,下载需要的版本即可
https://pytorch.org/get-started/previous-versions/这里有之前所有版本的pytorch,根据需要自己选择
然后复制下载的命令后,进入anaconda prompt,conda activate你刚才创建的环境名称(例子里面是pytorch),然后将下载命令输入回车
问你是否需要安装这些包,直接y确定
下载完成以后用pip list看看对应的包是否安装成功
输入python进入python模式,然后输入torch.cuda_is_available(),如果返回True,表示可以GPU使用该pytorch
pycharm安装
https://www.jetbrains.com/zh-cn/pycharm/
进入官网点击下载以后问你安装哪个版本的,肯定白嫖啊
然后安装就好了
配置pycharm
首先create new project
location根据自己情况定,然后编译器选择Existing interpreter,点后面那三个点
选择conda环境,然后下拉会自动填上你刚刚配置那个环境,找到对应的名字然后OK
然后就完成了pytorch的配置,如果你的python console不是In[2],In[3]这样的而是>>>这样的,需要给你的python环境加一个ipython,然后重启一下pycharm就好了