目录
1 下载并安装PyCharm,Anaconda
1.1 下载PyCharm
链接:https://www.jetbrains.com/pycharm/
1.2 下载Anaconda
清华镜像链接:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
官方链接:https://www.anaconda.com/products/individual
1.3 安装PyCharm
按照指引进行安装,可以更改安装目录
1.4 安装Anaconda
按照指引进行安装,可以更改安装目录;
just me改为all user;
1.5 配置系统环境变量
"此电脑"→"属性"→"高级系统设置"→"环境变量"→编辑"用户变量"的"path"→添加图中第2-5共四个环境变量→点击确定;
2 下载并安装Pytorch
2.1 匹配CUDA版本
桌面右键,打开NVIDIA控制面板,点击左下角系统信息,查看3D设置中NVCUDA64.DLL的产品名称,找到CUDA的版本,例如,下方图示的版本为11.1.106
打开Pytorch官网
链接:https://pytorch.org/get-started/locally/
选择pip安装,以及自己的CUDA版本,复制最下方的命令行,即Run This Command中的内容
2.2 下载并安装Pytorch
打开Anaconda Prompt,将上一步复制的内容粘贴至命令行,等待安装完毕即可
输入下列命令,若安装成功,则返回TURE,否则没有安装成功
python
import torch
import torchvision
2.3 在PyCharm中配置并使用Pytorch
新建一个项目
如果直接import torch,会出现如下提示,说明刚才安装的pytorch并没有被调用,因此需要采取措施
点击"File"→"Settings..."→"Project:自己的项目名称"→"Python Interpreter"→"右侧齿轮图标"→"add..."
点击Conda Environment,选择"Existing environment",勾选"Make available to all projects",选择右侧三个点的图标
选择自己的anaconda安装路径下的python.exe,确认并返回主界面
可以看到,此时已经不再报错,可以开始愉快的Pytorch之旅了
下次新建时,选择本次配置好的环境即可
3 测试样例
待续...