Pytorch | 学习笔记（一）

备注：个人电脑是WIN10

一、pytorch的安装

1.查看是否已安装 pytorch

从cmd（win+R）打开anaconda环境，

conda info -e （查看所有的虚拟环境）

activate -name（虚拟环境名字)（进入到该虚拟环境中）

pip list（查看安装好的库 )

2.安装CUDA和CUDNN

查看自己的cuda，去 table2 查找对应的

​

比如我的版本是CUDA 11.0.2 GA

确认好版本，下载对应的 CUDA 和 CUDNN，建议安装在C盘（下载CUDNN需要先注册）

3.安装pytorch

建议直接官网下载 pytorch

复制后，输入cmd，等待安装

4.检查安装是否成功

python

import torch 

torch.cuda.is_available()

不报错，则安装成功。如图。

5.调试环境（不必须）

如果想在jupyter notebook中使用配置好的anaconda环境的话，可以直接在环境中

conda install nb_conda

然后再进入jupyter notebook就可以了

如果想使用pycharm，在Files-Settings-Python Interpreter里面设置一下，就可以了。

二、简单函数

1.创建一个矩阵

x = torch.empty(5,3)

tensor是基本单位

2.创建一个随机矩阵

x = torch.rand(5,3)

3.初始化一个全零矩阵（5x3）

x = torch.zeros(5,3,dtype=torch.long)

​

 4.直接赋值

x = torch.tensor([5.5,3])

 5.展示矩阵大小

x = x.size()

６.view - 改变矩阵维度

x = torch.randn(4,4)
y = x.view(16)
z = x.view(-1,8)
print(x.size(),y.size(),z.size())
print(x)
print(y)
print(z)

 6.与 numpy 协作

import numpy as np
import torch
# torch -> numpy
a = torch.ones(5)
b = a.numpy()
print(b)

# numpy -> torch
a1 = np.ones(5)
b1 = torch.from_numpy(a1)
print(b1)

感谢以下博主提供的帮助：

Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_安装cudnn

使用anaconda虚拟环境运行Jupyter Notebook详解_w55100的博客-CSDN博客_anaconda jupyter 环境