学习pytorch
- 1. 环境安装
- 配置镜像源
- conda命令记录
- 图像相关代码
- 遇到的问题
- 1. torch.cuda.is_available() == False
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
