引言
笔者之前用的是Tensorflow+keras的环境,直接配置在anaconda的base环境下。由于研究需要,现打算转为pytorch环境。以此文记录一下整个过程以及常用指令。
CUDA11.4对应的torch版本
torch-1.10.1+cu113-cp38-cp38-win_amd64
使用torch.cuda.is_available()指令可以显示为True。
环境配置常用指令
conda create --name mytorch python=3.8
conda activate mytorch
conda install torch-1.10.1+cu113-cp38-cp38-win_amd64.whl
加入jupyter notebook的kernel环境
python -m ipykernel install --name mytorch
升级Anaconda Navigator
conda upgrade --all