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中实现对Timestamp和Datetime及UTC时间之间的转换
Apr 08 Python
探究Python多进程编程下线程之间变量的共享问题
May 05 Python
浅谈flask截获所有访问及before/after_request修饰器
Jan 18 Python
Python实现带参数与不带参数的多重继承示例
Jan 30 Python
对python中raw_input()和input()的用法详解
Apr 22 Python
关于python2 csv写入空白行的问题
Jun 22 Python
Python3使用TCP编写一个简易的文件下载器功能
May 08 Python
python、Matlab求定积分的实现
Nov 20 Python
pytorch forward两个参数实例
Jan 17 Python
Python3监控windows,linux系统的CPU、硬盘、内存使用率和各个端口的开启情况详细代码实例
Mar 18 Python
python中操作文件的模块的方法总结
Feb 04 Python
Pyhton模块和包相关知识总结
May 12 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
Discuz! 5.0.0论坛程序中加入一段js代码,让会员点击下载附件前自动弹出提示窗口
2007/04/18 PHP
php统计文件大小,以GB、MB、KB、B输出
2011/05/29 PHP
Yii结合CKEditor实现图片上传功能
2014/06/13 PHP
php中实现字符串翻转的方法
2017/02/22 PHP
JS 动态加载脚本的4种方法
2009/05/05 Javascript
jQuery 获取URL参数的插件
2010/03/04 Javascript
javascript setTimeout()传递函数参数(包括传递对象参数)
2010/04/07 Javascript
AngularJS入门教程之AngularJS表达式
2016/04/18 Javascript
jquery对所有input type=text的控件赋值实现方法
2016/12/02 Javascript
node.js与C语言 实现遍历文件夹下最大的文件,并输出路径,大小
2017/01/20 Javascript
jQuery读取XML文件的方法示例
2017/02/03 Javascript
ES6中Math对象新增的方法实例详解
2017/04/25 Javascript
Bootstrap模态框插入视频的实现代码
2017/06/25 Javascript
微信小程序之页面跳转和参数传递的实现
2017/09/29 Javascript
webpack+vue中使用别名路径引用静态图片地址
2017/11/20 Javascript
vue-cli3配置与跨域处理方法
2019/08/17 Javascript
vue form表单post请求结合Servlet实现文件上传功能
2021/01/22 Vue.js
[43:26]完美世界DOTA2联赛PWL S2 Forest vs Rebirth 第二场 11.20
2020/11/23 DOTA
深入浅析Python中join 和 split详解(推荐)
2016/06/30 Python
Windows平台Python连接sqlite3数据库的方法分析
2017/07/12 Python
python中将zip压缩包转为gz.tar的方法
2018/10/18 Python
PyCharm设置Ipython交互环境和宏快捷键进行数据分析图文详解
2020/04/23 Python
Python 捕获代码中所有异常的方法
2020/08/03 Python
python压包的概念及实例详解
2021/02/17 Python
使用CSS3实现字体颜色渐变的实现
2020/08/10 HTML / CSS
美国时装品牌:Nautica(诺帝卡)
2016/08/28 全球购物
菲律宾酒店预订网站:Hotels.com菲律宾
2017/07/12 全球购物
求职信需要的五点内容
2014/02/01 职场文书
2014年党员自我评价材料
2014/09/22 职场文书
2015年度培训工作总结范文
2015/04/02 职场文书
单身证明格式样本
2015/06/15 职场文书
2016大学生入党积极分子心得体会
2016/01/06 职场文书
Python基础之赋值,浅拷贝,深拷贝的区别
2021/04/30 Python
MongoDB数据库的安装步骤
2021/06/18 MongoDB
Go 语言下基于Redis分布式锁的实现方式
2021/06/28 Golang
Golang MatrixOne使用介绍和汇编语法
2022/04/19 Golang