Pytorch环境安装及配置
1、创建conda环境,名为pytorch
conda create -n pytorch python=3.10
2、在任务管理器的性能中确认显卡,是否支持CUDA。其次,确认显卡驱动,cuda9.2支持396.26以上的驱动,可以在命令行使用 nvidia-smi 来看自己驱动是否满足要求,如果低于396.26,可以使用各种管家更新驱动或者去官网下载最新驱动,https://www.nvidia.com/Download/index.aspx?lang=cn
3、去Pytorch官网根据自己的需求获取安装代码
*没有能用CUDA的英伟达显卡(去CUDA官网确认)则选CPU
4、在cmd进入激活pytorch环境后,运行安装代码
*这里先添加清华源,在把代码最后的-c pytorch去掉会安装得很快,具体见https://blog.csdn.net/zzq060143/article/details/88042075
*安装后用pip list确认已安装的package,可以看到torch在列表里。另一个方法是输入python,在python环境下输入import torch,不报错说明安装成功。如果安装了CUDA版本,继续输入torch.cuda.is_available(),返回True则成功
Python编辑器
以Pycharm为例,新建项目,选existing intepreter,找到之前创建的名为pytorch的环境。
如果要在pytorch环境下安装jupyter notebook:
conda install jupyter notebook
或者
conda install nb_conda_kernels
然后命令行输入jupyter notebook跳转网页
Package学习
dir()看package里有哪些分区和工具
help()看工具怎么用(说明书)