pytorch
syvge
CV
展开
-
RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected
(默认cuda版本和pytorch,以及nvidia-driver匹配)1. 首先进入python环境,通过torch.cuda.is_available()或者torch.cuda.device_count()来确认cuda是否安装,也可以在命令行里面使用nvcc -V来查看cuda是否安装2. 如果上述操作没问题,检查代码里面是否有这个:os.environ['CUDA_VISIBLE_DEVICES'] = '0'把自己电脑上的cudaid放到里面去...原创 2022-04-28 20:18:03 · 2071 阅读 · 0 评论 -
Python中无法导入上一级目录下的包
parentdir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))sys.path.insert(0,parentdir)比如你要导入的是当前Py文件上一级目录中的某个代码,则parentdir的获取就是使用dirname两次,然后将路径添加到系统中去...原创 2022-03-18 09:18:43 · 1400 阅读 · 0 评论 -
python init文件的作用
init.py文件的作用是将文件夹变为一个python模块。如果__init__.py文件不存在,这个目录就仅仅是一个目录,而不是包,就不能被导入或者包含其他模块和嵌套包。通常该文件为空,在我们导入一个包时,其实是导入了它的__init__文件,这样我们就可以在__init__中批量导入我们所需要的模块,而不需要一个个导入。我们可以在__init__.py中再导入其他的包或者模块,比如:import readersimport writersimport commandsimport users转载 2021-12-25 17:17:32 · 4020 阅读 · 0 评论 -
MeshiGrid画三维图
使用meshgrid生成网格t = np.linspace(1, 5, 5)x, y = np.meshgrid(t,t)使用plotsurface(z使用自定义的函数比如高斯分布生成)import numpy as npfrom mpl_toolkits.mplot3d import Axes3Dimport matplotlib.pyplot as pltfig = plt.figure()ax = Axes3D(fig)ax.plot_surface(x, y, z, .原创 2021-11-30 09:18:05 · 568 阅读 · 0 评论 -
Graphviz模型可视化
sudo apt-get install graphvizpip install graphvizcodingdef make_dot(vars, params = None): """ Produces Graphviz representation of PyTorch autograd graph Blue nodes are the Variables that require grad, orange are Tensors saved for backward .原创 2021-11-11 10:40:11 · 233 阅读 · 0 评论 -
matplotlib的颜色库
颜色:matplotlibcnames = [‘blue’,‘blueviolet’,‘brown’,‘burlywood’,‘cadetblue’,‘chartreuse’,‘chocolate’,‘coral’,‘cornflowerblue’,‘cornsilk’,‘crimson’,‘cyan’,‘aliceblue’,‘antiquewhite’,‘aqua’,‘aquamarine’,‘azure’,‘beige’,‘bisque’,‘black’,‘bl原创 2021-10-13 08:51:32 · 661 阅读 · 0 评论 -
Anoconda
设置默认不启动虚拟环境conda config --set auto_activate_base false原创 2021-06-06 23:59:57 · 182 阅读 · 0 评论 -
Pytorch对梯度进行rescale
继承torch.autograd.Function类,然后定义一个前向和反向函数,ctx是自动求导过程中用于存储属性的对象,比如ctx.save_for_backward。在foward中可以定义计算方程,但是本例子中只是对某一层的梯度进行rescale,所以对输入不做处理。class GradientRescaleFunction(torch.autograd.Function): @staticmethod def forward(ctx, input, weight):原创 2021-05-19 14:46:43 · 411 阅读 · 2 评论