学习pytorch

1. 环境安装

B站小土堆视频

配置镜像源

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

http://www.manongjc.com/detail/39-mtfsyjznbjcupcz.html

conda命令记录

conda create -n pytorch python=3.8
conda activate pytorch 
conda deactivate pytorch 
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch
conda install nb_conda # 安装jupyter-notebook 多个相关包
conda install --use-local pkg-name

图像相关代码

from torch.utils.data import Dataset
from PIL import Image
img_path=./path
img = Image.open(img_path)
img.show() # 显示照片
dir(img)   # 查看所有可用属性

遇到的问题

1. torch.cuda.is_available() == False