python之pytorch安装

pytorch官网

pytorch中文文档

pytorch离线包地址

PyTorch 是基于以下两个目的而打造的python科学计算框架：

• 无缝替换NumPy，并且通过利用GPU的算力来实现神经网络的加速。
• 通过自动微分机制，来让神经网络的实现变得更加容易

pytorch-GPU版本安装

GPU版本的安装必须建立在自身已经安装好cuda和cudnn的基础上

1. 查看自身的cuda版本和cudnn版本

#在命令行中输入
nvcc -V

2. 在创建的环境中在线安装（其他版本cuda具体命令参考pytorch官网）

#pip命令
pip3 install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

# conda+CUDA 10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

3. 或者离线进行安装（在pytorch离线包地址下载.whl离线包到任意目录下，然后在目录下运行以下命令）

pip install torch-1.5.0%2Bcu101-cp36-cp36m-win_amd64.whl

4. 测试安装是否完成，在命令行中输入python打开python，在python中输入以下命令进行确认

import torch
torch.cuda.is_available()
#返回值为true则表示已经安装成功

pytorch-CPU版本安装

1. 在创建的环境中在线安装（具体命令参考pytorch官网）

# 使用临时源进行安装
sudo pip3 install torch torchvision torchaudio -i https://pypi.mirrors.ustc.edu.cn/simple/ 

# 或者使用conda
conda install pytorch torchvision torchaudio cpuonly -c pytorch

2. 或者离线进行安装（在pytorch离线包地址下载.whl离线包到任意目录下，然后在目录下运行以下命令）

pip install torch-1.5.0%2Bcpu-cp36-cp36m-win_amd64.whl

3. 测试安装是否完成，在命令行中输入python打开python，在python中输入以下命令进行确认

import torch
#没有报错则安装成功