ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

ubuntu18.04LTS服务器用vituralenv安装和配置pytorch和tensorflow

2019-10-26 17:00:06  阅读:271  来源: 互联网

标签:LTS bin vituralenv pip3 torch pytorch cuda install ipython


$ python3 -m venv tf14
====输入例子====
$ vim ~/.bashrc #(添加如下行,可以设置bash默认启动)
$ source ~/tf14/bin/activate
$ pip3 install tensorflow-gpu==1.14
$ pip3 install ipython
$ ipython
import tensorflow
quit()
$ deactivate

$ python3 -m venv pytorch13
$ source ~/pytorch13/bin/activate
$ pip3 install torch===1.3.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable.html
$ pip3 install ipython
$ ipython
====输入例子====
import torch as t
x = t.rand(5,3)
y = t.rand(5,3)
if t.cuda.is_available():
x = x.cuda()
y = y.cuda()
print(x+y)
=====有结果代表成果====
tensor([[1.4504, 1.8244, 1.7167],
[0.6931, 1.4888, 0.5346],
[0.9387, 0.4503, 0.8335],
[1.3832, 0.9331, 1.4760],
[1.6562, 0.7950, 0.2785]], device='cuda:0')
quit()

$ deactivate

标签:LTS,bin,vituralenv,pip3,torch,pytorch,cuda,install,ipython
来源: https://www.cnblogs.com/jeshy/p/11743778.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有