一、安装Anaconda
二、安装pytorch
1 打开Anaconda prompt,创建pytorch环境
输入 conda create -n pytorch python=3.7
等待。。。
运行结束,出现proceed?
输入y
等待。。。
运行结束,创建pytorch环境成功
2 安装pytorch
输入nvidia-smi 查看支持版本信息
去PyTorch官网下载支持版本
进入pytorch环境:
输入 conda activate pytorch
复制相应版本安装信息:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
3 验证
通过导包验证安装是否成功
import torch
不报错就是安装成功
验证pytorch是否可以使用本机GPU
输入 torch.cuda.is_available()
三、安装pycharm
四、Pycharm配置pytorch环境
1 创建工程,选择环境
2 测试
五、jupyter使用pytorch
在安装Anaconda时,会附带直接安装了jupyter notebook,但是jupyter默认是在base环境中的,所以jupyter是无法直接使用pytorch的。
两种解决办法:
一是在base环境中在安装pytorch
二是在pytorch环境中安装jupyter(推荐)
1 在pytorch环境中安装jupyter
打开Anaconda prompt,切换环境到pytorch,安装:
打开jupyter notebook:
jupyter notebook中测试是否可用: