Anaconda配置pytorch-gpu虚拟环境的图文教程


Posted in Python onApril 16, 2020

1、更新NVIDIA驱动 

选对应自己显卡的驱动,(选studio版本,不要game版本)驱动链接 

Anaconda配置pytorch-gpu虚拟环境的图文教程

2、添加Anaconda清华镜像

方法一:anaconda命令替换

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --set show_channel_urls yes

(Mark)换回默认源代码:

conda config --remove-key channels

方法二:替换.condarc

Anaconda配置pytorch-gpu虚拟环境的图文教程

show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
custom_channels:
 conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

3.创建虚拟环境

创建:
conda create -n 环境名 python=X.X
开启:
activate 环境名
关闭:
conda deactivate
删除:
conda remove -n 环境名 --all
添加包:
conda install -n 环境名 包名
移除包:
conda remove -n 环境名 包名

4. 安装pytorch命令——对应情况自选:命令代码链接(pytorch官网)

1.PIP安装(推荐)

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

Anaconda配置pytorch-gpu虚拟环境的图文教程

这个速度比conda稳定 卡住了按回车好像还可以救回来

Anaconda配置pytorch-gpu虚拟环境的图文教程

2.conda安装(不推荐 老是中断)

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch (要去掉-c pytorch 不然还是默认源)
最终输入命令:
conda install pytorch torchvision cudatoolkit=10.1

Anaconda配置pytorch-gpu虚拟环境的图文教程

网络在各处中断 有时候48%又断了

Anaconda配置pytorch-gpu虚拟环境的图文教程

5.PIP安装完之后测试

import torch
flag = torch.cuda.is_available()
print(flag)
 
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

结果:被conda命令折腾一下午,终于被pip命令解救了!

Anaconda配置pytorch-gpu虚拟环境的图文教程

总结

到此这篇关于Anaconda配置pytorch-gpu虚拟环境步骤整理的文章就介绍到这了,更多相关Anaconda pytorch-gpu虚拟环境内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python中使用PIL库实现图片高斯模糊实例
Feb 08 Python
Python使用修饰器进行异常日志记录操作示例
Mar 19 Python
深入浅析python 协程与go协程的区别
May 09 Python
linux环境中没有网络怎么下载python
Jul 07 Python
用Python配平化学方程式的方法
Jul 20 Python
将matplotlib绘图嵌入pyqt的方法示例
Jan 08 Python
如何在django中运行scrapy框架
Apr 22 Python
Django ORM 查询表中某列字段值的方法
Apr 30 Python
浅谈多卡服务器下隐藏部分 GPU 和 TensorFlow 的显存使用设置
Jun 30 Python
详解python中的闭包
Sep 07 Python
改变 Python 中线程执行顺序的方法
Sep 24 Python
pytorch加载语音类自定义数据集的方法教程
Nov 10 Python
解决paramiko执行命令超时的问题
Apr 16 #Python
python nohup 实现远程运行不宕机操作
Apr 16 #Python
通过python 执行 nohup 不生效的解决
Apr 16 #Python
在python中使用nohup命令说明
Apr 16 #Python
VS2019+python3.7+opencv4.1+tensorflow1.13配置详解
Apr 16 #Python
python实现程序重启和系统重启方式
Apr 16 #Python
Mac PyCharm中的.gitignore 安装设置教程
Apr 16 #Python
You might like
php 根据url自动生成缩略图并处理高并发问题
2014/01/23 PHP
Smarty模板引擎缓存机制详解
2016/05/23 PHP
window.print打印指定div实例代码
2013/12/13 Javascript
Bootstrap每天必学之前端开发框架
2015/11/19 Javascript
div实现自适应高度的textarea实现angular双向绑定
2017/01/08 Javascript
Linux CentOS系统下安装node.js与express的方法
2017/04/01 Javascript
node实现基于token的身份验证
2018/04/09 Javascript
vue中倒计时组件的实例代码
2018/07/06 Javascript
JS面向对象之单选框实现
2020/01/17 Javascript
Python模块包中__init__.py文件功能分析
2016/06/14 Python
Python SqlAlchemy动态添加数据表字段实例解析
2018/02/07 Python
python+pandas分析nginx日志的实例
2018/04/28 Python
Python输出\u编码将其转换成中文的实例
2018/12/15 Python
实例讲解Python中整数的最大值输出
2019/03/17 Python
PIL图像处理模块paste方法简单使用详解
2019/07/17 Python
使用python分析统计自己微信朋友的信息
2019/07/19 Python
Python创建数字列表的示例
2019/11/28 Python
解决jupyter notebook 前面书写后面内容消失的问题
2020/04/13 Python
IDLE下Python文件编辑和运行操作
2020/04/25 Python
基于Python的Jenkins的二次开发操作
2020/05/12 Python
Python的轻量级ORM框架peewee使用教程
2021/02/05 Python
CSS3制作日历实现代码
2012/01/21 HTML / CSS
美国网上花店:JustFlowers
2017/02/12 全球购物
苏宁红孩子母婴商城:redbaby
2017/02/12 全球购物
在线购买廉价折扣书籍和小说:BookOutlet.com
2018/02/19 全球购物
GIVENCHY纪梵希官方旗舰店:高定彩妆与贵族护肤品
2018/04/16 全球购物
介绍一下Make? 为什么使用make
2016/07/31 面试题
高三自我鉴定怎么写
2013/10/19 职场文书
求职信的最佳写作思路
2014/02/01 职场文书
致裁判员加油稿
2014/02/08 职场文书
党风廉设责任书
2014/04/16 职场文书
不错的求职信范文
2014/07/20 职场文书
公务员岗前培训心得体会
2016/01/08 职场文书
本地通过nginx配置反向代理的全过程记录
2021/03/31 Servers
详解PHP用mb_string处理windows中文字符
2021/05/26 PHP
B站评分公认最好看的动漫,你的名字评分9.9,第六备受喜欢
2022/03/18 日漫