2022-6-14【pyTorch安装】

学习目标:

成功安装

学习内容:

  1. pyTorch安装
  2. 掌握 pyTorch基本知识

学习时间:

  • 4小时

学习产出:

1、Windows在Anaconda Prompt进行

  1. 标题查看现存虚拟环境
conda env list  

2022-6-14【pyTorch安装】_第1张图片

  1. 创建虚拟环境
    https://blog.csdn.net/banwancha/article/details/113802634
conda create -n env_name python==3.7(3.6-3.8)

2022-6-14【pyTorch安装】_第2张图片

  1. conda换源(清华源)
    conda config --set show_channel_urls yes
  2. 查显卡
    2022-6-14【pyTorch安装】_第3张图片

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
  1. 检验
#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)

你可能感兴趣的:(python,pytorch,人工智能,python)