安装pytorch
安装PyTorch
pytorch官网
https://pytorch.org/
创建一个环境 名为:pytorch
conda create -n pytorch python=3.9
激活;然后输入:conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
安装
最好离线安装
测试是否安装成功
torch.cuda.is_available()
是一个PyTorch库函数,用于检查当前系统是否支持CUDA(Compute Unified Device Architecture)加速。CUDA是由NVIDIA开发的并行计算平台和API模型,用于在GPU上加速计算任务。
当调用torch.cuda.is_available()
时,它会返回一个布尔值。如果返回True
,则表示当前系统支持CUDA加速;如果返回False
,则表示当前系统不支持CUDA加速或者没有正确配置CUDA环境。
使用CUDA加速可以显著提高深度学习模型的训练和推理速度,因为GPU在并行计算方面比CPU更加高效。因此,在使用PyTorch进行深度学习任务时,可以使用torch.cuda.is_available()
来检查是否可以利用GPU加速来提高计算性能。
Pytorch安装Jupyter
温馨提示:貌似出问题了,后期再解决
进入pytorch环境:
# 法1:conda install nb_conda_kernels # 法2:anaconda navigator
本文来自博客园,作者:__username,转载请注明原文链接:https://www.cnblogs.com/code3/p/17450410.html
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步