Python环境下的pytorch环境安装

---

Python环境下的pytorch环境安装以及学习资料

自己刚接触pytorch学习，做个学习笔记来记录一下学习过程

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、学习资源

1.视频入门教程：

目前只是在跟着这个视频学习
Deeplizard《Pytorch神经网络高效入门教程》中文字幕版.

二、环境安装

1.cuda版本确认以及更新

        在桌面右键鼠标打开NVIDIA控制面板，点击左下角的系统信息，再点击组件即可查看当前版本的cuda版本号

       由于我的cuda版本是10.0，但现在最新的pytorch1.6，只有对应的9.2\10.1\10.2的cuda版本，故需要安装10.1、10.2版本的cuda，可以通过CUDA历史版本合集.找到并安装你所需要的cuda版本。
（安装只需要安装全部组件即可，他会自动删除上一版本的cuda）

2.在Anaconda3的基础下安装pytorch

       首先进入PyTorch官网选择你所需要的版本，我的选择如下图所示：
       打开Anaconda Prompt,为了保证下载流畅，不丢（如果丢包或者下载卡顿，会停止当前安装）我试过很多方法后，最后选用了这个兄弟的方法成功安装：

首先输这个，更换镜像源（注意顺序，第四条一定要在最后，原因不详）

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/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

最后，运行执行安装pytorch（也就是官网最后给你的Run this commannd）

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

搞定

版权声明：黑体部分为CSDN博主「qq_34068584」的原创文章，遵循CC 4.0 by-sa版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_34068584/article/details/90112484

3.测试是否安装成功

输入1：

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

输出1：

tensor([[0.1258, 0.1577, 0.9321],
        [0.5674, 0.8834, 0.4103],
        [0.2404, 0.5602, 0.1996],
        [0.7360, 0.0567, 0.1273],
        [0.5201, 0.9893, 0.5491]])

输入2：（确认是否可用cuda）

import torch
torch.cuda.is_available()

输出2：

True

---

总结

花了一天时间终于配置好了环境加油！