1.创建环境
自行下载Anaconda
也可以下载下面这个miniAnaconda,这个会更方便
链接:https://pan.baidu.com/s/1yKbEywfEsAioL6-YUv3xsw
提取码:a4h1
用 conda 创建问文件夹PT,在python3.6的环境下。
指令如下:
conda create -n PT python=3.6
之后,弹出提示,输入 y,即可安装。
安装成功后,输入以下指令:
conda activate PT
进入 PT 环境,可以看到左边的 base 变成了 PT。
**
2. 安装 PyTorch 准备
conda info查看有没有安装清华源,如果没有,则添加,可以加速下载
**
首先安装清华源
**
添加清华源
不清楚哪个有效,可以都下载
以下两条是Anaconda官方库的镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
以下是Anaconda第三方库 Conda Forge的镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
以下两条是Pytorch的Anaconda第三方镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
安装 PyTorch
打开官网https://pytorch.org/
下拉到下面的页面。
按照上图选择 ,如果需要CUDA按照自己的显卡进行选择
进入PT环境,复制这一段代码,这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包,因此这里用命令行:
conda install pytorch torchvision cpuonly
接着等待安装成功就好了
最后可以通过命令行验证一下,可以看到已经正确安装PyTorch。
**
3.验证是否安装成功
**
(1)在命令行左边为 PT 环境中,输入 python
(2)之后,输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了
(3)接下来,输入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。
4.使用
(1)jupyter notebook 使用(如果有多个环境需要配置才可使用pytorch)
配置过程:
进入 PT 环境 ,安装ipykernel
conda install ipykernel
安装成功,将环境写入kernel
python -m ipykernel install --user --name PT --display-name “torch(python3.6)”
随后打开 jupyter notebook 选中torch(python3.6) 创建文件,如下图
成功
(2)pycharm 使用
new project
修改后点击create
之后点击 file—>setting—>project interpreter
install torch 成功后测试
成功
参考:https://www.cnblogs.com/zhouzhiyao/p/11784055.html
https://blog.csdn.net/qq_37717494/article/details/105568350