Pytorch
框架下载
针对电脑4060RTX
显卡
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
检验Pytorch
安装成功,能连接GPU
import torch
print(torch.cuda.is_available())
# 显示True才行
激活pytorch
:
在cmd中输入 conda activate pytorch
# 直接进入jupyter notebook使用pytorch
在pytorch环境中可输入jupyter notebook