安装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
posted @ 2023-06-01 22:17  __username  阅读(10)  评论(0编辑  收藏  举报

本文作者:DIVMonster

本文链接:https://www.cnblogs.com/guangzan/p/12886111.html

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。