首页 > 其他 > 详细

conda配置镜像并安装pytorch

时间:2021-07-26 22:35:13      阅读:18      评论:0      收藏:0      [点我收藏+]

一、删除镜像

# 换回默认源(清除所有用户添加的镜像源路径,只保留默认的路径)
conda config --remove-key channels
# 显示所有镜像通道路径命令
conda config --show channels

二、添加镜像

conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/

三、创建环境

conda create --name pt python=3.7

四、安装

conda install pytorch torchvision torchaudio cudatoolkit=10.1

 五、验证

import torch
print(torch.__version__)
print("gpu:", torch.cuda.is_available())

 

conda配置镜像并安装pytorch

原文:https://www.cnblogs.com/zhangxianrong/p/15061802.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!