学习pytorch1 环境安装

1. 环境安装

B站小土堆视频

安装anaconda

安装pycharm

配置镜像源

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
conda install pytorch-cpu torchvision-cpu -c pytorch # 为了引入torch.nn.functional

图像相关代码

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

说是cuda版本不一致，暂未解决
搜索到的暂未尝试

1、首先确定显卡是否支持CUDA，https://www.geforce.com/hardware/technology/cuda/supported-gpus 网址打不开了
2、确保驱动升级到最新
nvidia-smi 来看自己驱动是否满足要求，https://www.nvidia.com/Download/index.aspx?lang=cn 官网下载最新驱动
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
在选择显卡时的那一步把1改成0
py3.8.+pytorch是11.3