学习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

遇到的问题

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