学习目标:
成功安装
学习内容:
- pyTorch安装
- 掌握 pyTorch基本知识
学习时间:
- 4小时
学习产出:
1、Windows在Anaconda Prompt进行
- 标题查看现存虚拟环境
conda env list
- 创建虚拟环境
https://blog.youkuaiyun.com/banwancha/article/details/113802634
conda create -n env_name python==3.7(3.6-3.8)
- conda换源(清华源)
conda config --set show_channel_urls yes - 查显卡
5、离线下载
下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
通过上面下载地址,我们需要下载好对应版本的pytorch和 torchvision 包,然后打开Anaconda Prompt/Terminal中,进入我们安装的路径下。
cd package_location
conda activate env_name
接下来输入以下命令安装两个包
conda install --offline pytorch-1.4.0-py3.8_cuda92_cudnn7_0.tar.bz2
conda install --offline torchvision-0.6.1-py38_cu92.tar.bz2
在线安装
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
- 检验
#print('emo')
#矩阵0-1
import torch
x = torch.rand(4,3)
print(x)
#0矩阵
import torch
x = torch.zeros(4,3,dtype = torch.long)
print(x)
#向量
import torch
x = torch.tensor([5.5,3])
print(x)
x = x.new_ones(4, 3, dtype = torch.double)
print(x)
x = torch.randn_like(x,dtype = torch.float)
print(x)
print(x.size())
print(x.shape)