Advertisement

在Ubuntu上设置Pytorch环境

阅读量:

CPU版本安装指南

https://www.cnblogs.com/wangqinze/p/13407610.html

完成后检验是否有误

复制代码
    python
    import torch
    import torchvision
    
    
      
      
      
    

若未出现任何错误信息,则表示操作已成功结束。

GPU版本安装指南

在完成CUDA与cudnn的配置之后,可直接采用官方提供的命令进行操作:

复制代码
    conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
    
    
      
    
在这里插入图片描述

遇到以下错误提示时:
CondaHTTPError: 在访问 <https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/linux-64/current_repodata.

全部评论 (0)

还没有任何评论哟~