前言
Pytorch的安装需要Python的支持,借助Anaconda软件。
Anaconda的安装和配置参考
https://blog.csdn.net/weixin_49732545/article/details/119533129
换清华源。参考https://www.cnblogs.com/yikemogutou/p/11396045.html
Pytorch安装
创建环境
conda create -n pytorch python=3.5.2
其中pytorch问环境名称可更改,python=3.5.2为该环境python版本,可更改。
激活环境
activate pytorch
安装方式一
进入Pytorch官网https://pytorch.org/
其中:PyTorch Build为版本
Your OS为你的操作系统
Package 安装方式 通常选conda或者pip
Compute Platform 安装GPU版本是需要选择cuda,安装CPU选择cpu
Run this Command: 复制该条命令进入激活后的环境
安装方式二(解决安装速度慢)
方式一安装时速度可能会很慢。
根据自己情况选择PyTorch Build、Your OS、Language、Compute Platform。其中Package选择pip。
Run this Command:里面的网址如:https://download.pytorch.org/whl/torch_stable.html,复制新建页面打开。
在里面找到对应的torch版本和torchvision版本,然后下载。
其中python版本要选择与安装时环境一致的版本。
注意!!!
网站下载要不pip命令中一直
注意版本,python环境,系统环境和cpu版还是gpu版。
如gpu版本需要cudatoolkit。
安装
在激活后的环境执行
conda install E:\download\cudatoolkit-11.0.221-h74a9793_0.conda
安装cudatoolkit。gpu版本时需要安装。其中E:\download\为存放该文件的路径。
执行
pip install E:\download\torch-1.7.1+cu110-cp38-cp38-win_amd64.whl -i https://pypi.douban.com/simple
安装torch。E:\download\为存放文件路径。i https://pypi.douban.com/simple 为源 必要时下载安装相关联的包。
torchvision安装同torch。
测试
在命令行输入python进入
输入
import torch
如果没有报错,意味着 PyTorch 已经顺利安装了。
如果gpu版本
torch.cuda.is_available
如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。