安装Ubuntu20.04与安装NVIDIA驱动的教程

安装Ubuntu 20.04 安装NVIDIA 驱动 配置Pytouch 和tensorflow环境本机环境：戴尔G3 3579win10 ，系统在128固态硬盘安装Ubuntu20.04

安装NVIDIA驱动

重启后，⿇事情没有。搭建pytouch安装pycharm

点击tools->create desktop entry 直接⽣成快捷键设置编译器为pytouch

填写代码测试使⽤了GPU：

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()) # True# cuda:0

# GeForce GTX 1060

# tensor([[0.5772, 0.5287, 0.0946],# [0.9525, 0.7855, 0.1391],

# [0.6858, 0.5143, 0.8188]], device='cuda:0')

安装tensorflow14

import tensorflow as tf

from tensorflow.python.client import device_libprint(device_lib.list_local_devices())import warnings

warnings.filterwarnings(\"ignore\")

hello=tf.constant(\"Hello,Tensorflow\")print(hello)

a=tf.constant([1.0,2.0]) #定义常数b=tf.constant([3.4,4.0])result1=a+b

print(\"a+b=\

c=tf.constant([[3.0],[1.4]])result2=a+c

sess=tf.Session()

print(\"result1:\显⽰结果是“add:0\"的张量，shape只有⼀个元素，即维度是1# 2表⽰第⼀个维度有两个元素，且是浮点型try:

print(sess.run(result1)) print(\"result2:\ print(sess.run(result2)) print(sess.run(hello))except: #异常处理

print(\"Exception\")finally:

#关闭会话，释放资源 sess.close()

总结

到此这篇关于安装Ubuntu20.04与安装NVIDIA驱动的教程的⽂章就介绍到这了,更多相关安装Ubuntu20.04 NVIDIA驱动内容请搜索以前的⽂章或继续浏览下⾯的相关⽂章希望⼤家以后多多⽀持！