PyTorch 安装
- 本机环境
打开设备管理器,找到相应的显卡,查看本机是否支持GPU,如下图所示,这可以确定需要安装PyTorch 是GPU 还是CPU 版本
- Anaconda 安装
Anaconda官网地址:https://www.anaconda.com/distribution/,选择和机器环境对应的版本并下载安装
- PyTorch 安装
PyTorch 主页:https://pytorch.org/
PyTorch 主要使用Pip 或 Conda安装,这里使用Anaconda作为安装工具,选择CUDA版本
打开Anaconda,创建一个新的环境,指定Python的版本
打开Anaconda Prompt,可以查看当前的环境或者切换环境:
列出环境:conda info --env
切换环境:conda activate My_PyTorch
查看安装的包:conda list
使用Pip方式安装,命令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113,执行过程如下
- 验证PyTorch是否安装成功
1:打开python:python
2:测试pytorch:import torch
如下图所示,没有遇到问题,则表示成功了
- 安装Jupyter
安装Anaconda的同时,会在base环境安装jupyter notebook,因此并不能在此jupyter notebook运行torch包,在前面新建的环境中运行命令
conda config --set remote_read_timeout_secs 1000.0
conda install nb_conda
完成后,执行jupyter notebook,如下图,选择之前创建的conda 环境
输入import torch
,点击运行
再输入torch.cuda.is_available()
输出 True
,表示我们在自己创建的pytorch环境中安装好了jupyter包