win10安装pytorch
安装cuda10(官网支持的是8.0,9.0,10.0,由于之前安装的是9.2,结果安装好后导入错误,遂安装了最新的cuda10),安装方法直接去官网下载exe文件即可。 安装cudnn7.3,去官网下载cudnn,按照官网安装方法安装(拷贝几个文件即可)。 安装pytorch,按照官网方法安装即可(选择对应的版本,pip安装)。 测试:import torch没有报错即成功;print(torc… 阅读更多 »win10安装pytorch
安装cuda10(官网支持的是8.0,9.0,10.0,由于之前安装的是9.2,结果安装好后导入错误,遂安装了最新的cuda10),安装方法直接去官网下载exe文件即可。 安装cudnn7.3,去官网下载cudnn,按照官网安装方法安装(拷贝几个文件即可)。 安装pytorch,按照官网方法安装即可(选择对应的版本,pip安装)。 测试:import torch没有报错即成功;print(torc… 阅读更多 »win10安装pytorch
在使用eager模式的时候,如果类继承的是tf.keras.layers.Layer,要获得该类的变量,即类.variables,需要在类中使用self.add_variable()生成变量,一般可以在build函数中添加;如果继承的是tf.keras.Model,任意位置使用tf.get_variable()生成变量,并赋值给self.变量名。一般在__init__()函数中 ,如果使用tf.l… 阅读更多 »tensorflow学习(2)