环境
我这里用的是Anaconda最新的版本24.9.2,创建的pyton版本是3.9,虚拟环境名叫pytorch,下载的pytorch版本是12.1
初步配置
创建虚拟环境
首先进入Anaconda的终端,然后运行conda create -n xxx python=3.9
创建一个虚拟环境,这里xxx是虚拟环境的名字,后面跟着的是python的版本
然后用conda activate xxx
激活环境,退出环境用conda deactivate xxx
如果需要删除环境,用conda rm -n xxx --all
删除
下载pytorch
然后进入pytorch的官网,下滑找到自己需要的pytroch版本,然后复制出现的命令在conda对应的虚拟环境中进行下载 (漫长的等待...这里由于网络问题会下载失败,重试几次即可,注意有显卡的不要到网上去换镜像源,新版本的pytorch用镜像源下载不成功,是CPU版本的)
下载完成后进入python环境,然后import torch
导入torch库,再用torch.cuda.isavailiable()
查看是否正确安装好pytorch,如果安装成功,有显卡的出现true,没有的出现false
配置pycharm
这里我下载的是最新的pycharm,进入pycharm之后需要自己配置环境,新建一个项目需要获取Anaconda的环境,如下所示:
如果没有自动弹出,选择自定义环境后选择现有,类型选择conda,然后路径到conda的目录中的Liabrary的bin的conda.bat,就可以自动配置环境了,要注意的是最下面的环境要选择对应的虚拟环境!!!比如前面创建的是pytorch就选择pytorch,不能默认!!!!,之后创建项目就可以了
同样的,进入项目后在终端中import torch
导入torch库,再用torch.cuda.isavailiable()
查看是否正确安装好pytorch,如果安装成功,有显卡的出现true,没有的出现false
安装Jupyter Notebook
在新版的pytorch和python中,直接用conda install nb_conda
是不行的,要用conda install nb_conda_kernels
下载,完成之后输入jupyter notebook
就可以进入界面,然后创建一个新的jupyter book项目即可
同样的,我们在book中import torch
导入torch库,再shift + enter
后,用torch.cuda.isavailiable()
,然后按下shift + enter
查看是否正确安装好pytorch,如果安装成功,有显卡的出现true,没有的出现false
这里shift+enter
的作用是切换到下一个代码块并且运行上一个代码块,如果没报错说明成功