在Ubuntu上设置Pytorch环境
发布时间
阅读量:
阅读量
CPU版本安装指南
完成后检验是否有误
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)
还没有任何评论哟~
