pytorch安装

1. 安装 Anaconda

https://blog.csdn.net/qq_45344586/article/details/124028689

2. 检查显卡，更新驱动

• 查看显卡型号

  
  
  image.png

• 根据显卡型号查看cuda安装版本
  CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构（ISA）以及GPU内部的并行计算引擎。
  1.显卡驱动安装
  2.cuda安装
  3.cudann安装

详细步骤：
CUDA安装_天青色等烟雨blog的博客-CSDN博客_cuda安装

3. 创建PyTorch环境

conda  create  --name  pytorch python=3.5 numpy scipy # 创建指定python版本下包含某些包

4. 配置清华镜像源
conda config --show-sources #查看当前使用源
conda config --remove channels 源名称或链接 #删除指定源
conda config --add channels 源名称或链接 #添加指定源

conda config --add channels  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

5. 安装 PyTorch
PyTorch是一个开源的Python机器学习库，其前身是2002年诞生于纽约大学 的Torch。它是美国Facebook公司使用python语言开发的一个深度学习的框架，2017年1月，Facebook人工智能研究院（FAIR）在GitHub上开源了PyTorch。

1.进入pytorch虚拟环境

activate pytorch

2.安装
(22条消息) Pytorch最全安装教程（一步到位）_weiAweiww的博客-CSDN博客_pytorch安装

6. 测试
进入python环境中
输入一下代码

import torch 
print(torch.__version__)
print("gpu", torch.cuda.is_available())

输出如图所示表示安装成功

7. 安装jupyternotebook

jupyter notebook可能找不到虚拟环境
解决方法
(22条消息) 解决JUPYTER NOTEBOOK无法找到虚拟环境的问题_seaside2003的博客-CSDN博客_jupyter notebook找不到虚拟环境